CCCC - 天梯赛 -L3-003. 社交集群 并查集

L3-003. 社交集群

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

输入格式：

输入的第一行给出正整数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

题解：

        这道就是普通的并查集，不过中间使用了map进行优化。因为我需要从不同的并查组中得到更多关于组的性质。我就使用了map大大优化了查找的速度

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

int fa[1007];

int find1(int x){
    return x == fa[x]?x:find1(fa[x]);
}

void unit(int x,int y){
    int v = find1(x),u = find1(y);
    if (v < u) fa[u] = v;
    else       fa[v] = u;
    return ;
}

bool cmp(int x,int y){
    return x > y;
}

int vis[1005];
int a[1007];

int main()
{
    //ios::sync_with_stdio(false);

    for (int i = 0;i <= 1005;i ++)fa[i] = i;

    int n;
    cin >> n;

    for (int i = 1;i <= n;i ++){
        int k;
        scanf("%d:",&k);
        cin >> a[i];
        vis[a[i]] = 1;
        for (int j = 1;j <= k-1;j ++){
            int t;
            cin >> t;
            unit(t,a[i]);
            vis[t] = 1;
        }
    }

    map<int,int> se;
    map<int,int>::iterator it,it2;
    for (int i = 1;i <= n;i ++){
        int x = find1(a[i]);
        se[x] ++;
    }

   // it2 --;
    cout << se.size() << endl;
    vector<int> vp;
    for (it = se.begin();it != se.end();it ++){
        vp.push_back(it->second);

    }
    sort(vp.begin(),vp.end(),cmp);

    for (int i = 0;i < vp.size();i ++){
        cout << vp[i] << (i == vp.size()-1?'\n':' ');
    }
}