本题明显是一道并查集题目,如果直面硬刚选择连接人员并查集,那么会感觉较为麻烦。
一种感觉更轻松的做法是连接爱好,这样做本题便会变成一个简单的带权并查集模板题。
并且答案要求给的是爱好合并成的群,没有问任何与人员有关的信息,而且完全可以ac。
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int f[N], size[N];
int find(int x){
return x==f[x]?x:f[x] = find(f[x]);
}
int main(){
// freopen("in.txt", "r", stdin);
int n;
scanf("%d", &n);
for(int i = 1; i<= 1000; i++) {
f[i] = i;
size[i] = 0;
}
while(n--){
int k, x, fx, y, fy;
scanf("%d: %d", &k, &x);
for(int i = 1; i <= k-1; i++){
scanf("%d", &y);
fx = find(x);
fy = find(y);
if(fx != fy){
f[fy] = fx;
size[fx] += size[fy];
}
}
fx = find(x);
size[fx]++;
}
int cnt = 0;
priority_queue<int> q;
for(int i = 1; i <= 1000; i++){
if(size[i] && f[i] == i) {
cnt++;
q.push(size[i]);
}
}
printf("%d\n", cnt);
int flag = 0;
while(q.size()){
if(flag) printf(" ");
if(flag == 0){
flag = 1;
}
printf("%d", q.top());
q.pop();
}
return 0;
}