L3-003 社交集群 (30 分) （并查集）

题目链接

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

输入格式：

输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：

其中K​i(>0)是兴趣爱好的个数，hi[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 <iostream>
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-9
#define x first
#define y second
const int mod = 1e9 + 7;
const int MOD = 1e4+7;
const int N = 1e6 + 10;
const int M = 1111;
int dx[]={-1, 0, 1, 0};
int dy[]={0, 1, 0, -1};
int dxy[][2]={{0,1},{1,0},{1,1},{-1,1}};
using namespace std;

int vis[M];
int dp[M];
map<int,int>mp;
vector<int>vp;

int Find(int x){
    if(x!=vis[x]) vis[x]=Find(vis[x]);
    return vis[x];
}

void Merge(int u,int v){
    int x=Find(u);
    int y=Find(v);
    if(x!=y) vis[y]=x;
}

void init(){
    for(int i=0;i<=M;i++) vis[i]=i;
}

void solve(){
    init();
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        int k;
        char c;
        cin>>k>>c>>dp[i];
        k--;
        while(k--){
            int x;
            cin>>x;
            Merge(dp[i],x);
        }
    }
    for(int i=0;i<n;i++){
        int x=Find(dp[i]);
        if(!mp.count(x)) mp[x]=1;
        else mp[x]++;
    }
    for(int i=1;i<=1000;i++){
        if(mp.count(i)){
            vp.pb(mp[i]);
        }
    }
    sort(vp.begin(),vp.end());
    int len=vp.size();
    cout<<len<<endl;
    for(int i=len-1;i>=0;i--){
        if(i==0) cout<<vp[i]<<endl;
        else cout<<vp[i]<<" ";
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    solve();
	return 0;
}