PAT 甲级 1107 Social Clusters (30分)(并查集)

思路:

1.先用并查集算出所有交友圈集合;
2.再挨个遍历每个用户,查询他在哪个集合,相应集合数量++,最后数量倒序输出即可;

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=1005;
int n,par[maxn],rnk[maxn],tag[maxn],peo[maxn];
void init_(){
	for(int i=0;i<maxn;i++) par[i]=i;
}
int find(int x){
	if(par[x]==x) return x;
	else return par[x]=find(par[x]);
}
void unite(int x,int y){
	x=find(x); y=find(y);
	if(x==y) return;
	if(rnk[x]<rnk[y]) par[x]=y;
	else{
		par[y]=x;
		if(rnk[x]==rnk[y]) rnk[x]++;
	}
}
void solve(){
	map<int,int> cnt;
	for(int i=1;i<=n;i++) cnt[find(peo[i])]++;
	vector<int> v;
	for(auto e:cnt) v.push_back(e.second);
	sort(v.begin(),v.end(),greater<int>());
	cout<<v.size()<<'\n'<<v[0];
	for(int i=1;i<v.size();i++) cout<<' '<<v[i];
}
int main(){
	init_();
	cin>>n;
	for(int i=1;i<=n;i++){
		int k; cin>>k; getchar();
		int x,y; cin>>x; tag[x]=1; peo[i]=x;
		for(int i=1;i<k;i++){
			int y; cin>>y; tag[y]=1; unite(x,y); x=y;
		}
	}
	solve();
	return 0;
}
posted @ 2019-12-05 12:56  YuhanのBlog  阅读(77)  评论(0编辑  收藏  举报