L3-003 社交集群 (30 分)（注释详解）

并查集

类似的并查集问题

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

const int N = 1e5;


int f[N],a[N];
int n;
vector<int> ans,v[N],res;
set<int> st;


int find(int x){
	if(x != f[x]) f[x] = find(f[x]);
	return f[x];
}

void Union(int a,int b){
	int pa = find(a);
	int pb = find(b);
	
	if(pa != pb){
		f[pa] = pb;
	}
	
	
}


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



int main(){
	init();
	scanf("%d",&n);
	
	
	for(int i=1;i<=n;i++){
		int m,un;
		scanf("%d:",&m);
		scanf("%d",&un);
		v[i].push_back(un);
		for(int j=0;j<m-1;j++){
			int t;
			scanf("%d",&t);
			Union(un,t);
		}
	}
	

	
		
		for(int i=1;i<=n;i++){
			//通过根节点统计每个集群的人数 
			a[find(v[i][0])]++;
			//存放根节点 
			st.insert(find(v[i][0]));
		}
		
	//集群的数目 
	cout<<st.size()<<endl;
	
	set<int> :: iterator it = st.begin();
	
	int cnt = 0;
	
	//存放每个集群的人数 
	for(it;it!=st.end();it++){
		res.push_back(a[*it]);
	}
	
	//降序排序 
	sort(res.begin(),res.end(),greater<int>());
	
	cout<<res[0];
	
	for(int i=1;i<res.size();i++)
		cout<<" "<<res[i];  
	
	
	
	return 0;
}