L3-003. 社交集群

Posted 2020-10-15 给杰瑞一块奶酪~

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入格式：

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

K_i: h_i[1] h_i[2] ... h_i[K_i]

其中K_i（>0）是第i个人的兴趣的数量，h_i[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出格式：

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4

输出样例：

3
4 3 1


并查集 每次对有某一个爱好所有人扫一遍 有相同的就并一下，最后检查  输出。

代码：

#include <iostream>
#include <map>
#include <queue>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define Max 1001

using namespace std;

int mp[Max][Max];
int n;
int f[Max],num[Max];
int *p[Max],c;
bool cmp(int *a,int *b)
{
    return *a > *b;
}
void init()///初始化
{
    for(int i = 1;i <= n;i ++)
        f[i] = i;
}
int getf(int x)///寻找父亲
{
    if(f[x] != x)return getf(f[x]);
    return x;
}
void marriage(int x,int y)///合并
{
    int xx = getf(x);
    int yy = getf(y);
    f[yy] = xx;
}
int main()
{
    int k,d;
    cin>>n;
    init();
    for(int i = 1;i <= n;i ++)
    {
        cin>>k;
        cin.get();///读取 ‘:‘
        while(k --)
        {
            cin>>d;///每个d都扫一遍
            for(int j = 1;j <= n;j ++)
                if(mp[j][d])marriage(j,i);
            mp[i][d] = 1;
        }
    }
    for(int i = 1;i <= n;i ++)
    {
        d = getf(i);
        if(d == i)p[c ++] = &num[i];///如果父亲是自己就加入p序列 num记录每个群体个数
        num[d] ++;///计算个数
    }
    sort(p,p+c,cmp);
    cout<<c<<endl<<*p[0];
    for(int i = 1;i < c;i ++)
        cout<<‘ ‘<<*p[i];
}