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;
}