L3-003 社交集群（并查集）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/leekerian/article/details/88806132

对于兴趣爱好我们只要加上1111就能够与人区分开了，然后用并查集维护，路径压缩一发

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;
int num[11111];

vector<int> vt;
vector<int> vt2;
vector<int> ans;
map<int,int> mp;
bool cmp(int a,int b)
{
    return a>b;
}
int find(int x)
{
    if(num[x]!=x) num[x]=find(num[x]);
    return num[x];
}


void un(int a,int b)
{
    int x=find(a);
    int y=find(b);
    if(x>y) num[y]=x;
    else num[x]=y;
}


int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        num[i]=i;
    for(int i=1;i<=n;i++)
    {
        int m;
        scanf("%d:",&m);
        vt.push_back(i);
        for(int j=0;j<m;j++)
        {
            int k;
            scanf("%d",&k);
            k+=1111;
            if(!num[k])
                num[k]=k;
            vt.push_back(k);
            un(num[i],num[k]);
        }
    }
    for(int i=0;i<vt.size();i++)
    {
        num[vt[i]]=find(num[vt[i]]);
        if(num[vt[i]]==vt[i]&&!mp[vt[i]])
        {
            vt2.push_back(vt[i]);
            mp[vt[i]]=1;
        }
    }
    for(int i=0;i<vt2.size();i++)
    {
        int k1=0;
        for(int j=0;j<vt.size();j++)
        {
            if(num[vt[j]]==vt2[i]&&vt[j]<=1000)
                k1++;
        }
        ans.push_back(k1);
    }
    sort(ans.begin(),ans.end(),cmp);
    cout<<ans.size()<<endl;
    for(int i=0;i<ans.size();i++)
        if(!i) printf("%d",ans[i]);
        else printf(" %d",ans[i]);
    cout<<endl;
    return 0;
}