L3-1 社交集羣(並查集)

L3-1 社交集羣（30 分）
當你在社交網絡平臺註冊時，一般總是被要求填寫你的個人興趣愛好，以便找到具有相同興趣愛好的潛在的朋友。一個“社交集羣”是指部分興趣愛好相同的人的集合。你需要找出所有的社交集羣。

輸入格式：

輸入在第一行給出一個正整數 N（≤1000），爲社交網絡平臺註冊的所有用戶的人數。於是這些人從 1 到 N 編號。隨後 N 行，每行按以下格式給出一個人的興趣愛好列表：

K
​i
​​ : h
​i
​​ [1] h
​i
​​ [2] … h
​i
​​ [K
​i
​​ ]

其中K
​i
​​ (>0)是興趣愛好的個數，h
​i
​​ [j]是第j個興趣愛好的編號，爲區間 [1, 1000] 內的整數。

輸出格式：

首先在一行中輸出不同的社交集羣的個數。隨後第二行按非增序輸出每個集羣中的人數。數字間以一個空格分隔，行末不得有多餘空格。

輸入樣例：

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<algorithm>
#include<stdio.h>
#include<iostream>
#include<set>
#include<vector>

using namespace std;
int f[1007];
int num[1007];
void init()
{
    for(int i =0;i<=1000;i++)
    {
        f[i] = i;
    }
}
int GetF(int x)
{
    int r = x;
    while(x != f[x])
        x = f[x];
    while(r != x)
    {
        int tmp = f[r];
        f[r] = x;
        r = tmp;
    }
    return x;
}
void Union(int a,int b)
{
    int p = GetF(a);
    int q = GetF(b);
    if(p != q)
    {
        f[q] = p;
    }
}
int main()
{
    int n;
    init();
    scanf("%d",&n);
    set<int> qq[n+1];
    for(int i =1;i<=n;i++)
    {
        int k;
        scanf("%d: ",&k);
        int pre = -1;
        while(k--)
        {
            int id;
            scanf("%d",&id);
            qq[i].insert(id);//把每個人的興趣加進來
            qq[0].insert(id);//存所有興趣
            if(pre!=-1)
            {
                Union(pre,id);
            }
            pre = id;
        }
    }
    set<int>::iterator it = qq[0].begin();
    int cnt =0;
    while(it != qq[0].end())
    {
        int ff = GetF(*it);
        if(*it == f[*it])
            cnt++;

        it++;
    }
    for(int i =1;i<=n;i++)
    {
        int k = *qq[i].begin();
        int p = GetF(k);
        num[p]++;
    }
    cout<<cnt<<endl;
    sort(num,num+1001);
    int fir =0;
    for(int i = 1000;i>=0;i--)
    {
        fir == 0? cout<<num[i]:cout<<" "<<num[i];
        fir =1;
        cnt--;
        if(cnt ==0)
            break;
    }
    cout<<endl;
}