L3-003 社交集群

当你在社交网络平台注册时，一般总是被要求填写你的个人兴趣爱好，以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式：

输入在第一行给出一个正整数 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<bits/stdc++.h>
#define maxsize 1002
using namespace std;
int pre[maxsize];///并查集
int ss[maxsize];///记录汇总
int s[maxsize];///记录集群人数
set<int>st;
int n;
///查找
int Find(int x){
    if(pre[x]==x)
        return x;
    else
        return pre[x]=Find(pre[x]);///带路径压缩
}
///合并
void Union(int x,int y){
    int xx=Find(x);
    int yy=Find(y);
    if(xx<yy)
        pre[yy]=xx;
    else
        pre[xx]=yy;
}
int main(){
    cin>>n;
    for(int i=1;i<maxsize;i++)
        pre[i]=i;
    for(int i=1;i<=n;i++){
        int m,t,tt;
        scanf("%d: %d",&m,&t);
        for(int j=0;j<m-1;j++){
            cin>>tt;
            Union(t,tt);
        }
        ss[i]=Find(t);
    }
    for(int i=1;i<=n;i++){
        int k=Find(ss[i]);
        if(st.count(k))
            s[k]++;
        else{
            s[k]=1;
            st.insert(k);
        }
    }
    cout<<st.size()<<endl;
    set<int>::iterator it;
    vector<int>v;
    ///集群汇总
    for(it=st.begin();it!=st.end();it++)
        v.push_back(s[*it]);
    sort(v.begin(),v.end(),greater<int>());
    for(int i=0;i<v.size();i++)
        if(i)
            cout<<" "<<v[i];
        else
            cout<<v[i];
    return 0;
}

执行情况：