考察并查集
题目大意:
有N个人,每个人喜欢若干项活动,如果两个人有任意一个活动相同,那么就称他们处于同一个社交网络(A,B属于一个社交网络,B,C属于一个社交网络,则ABC属于同一个社交网络)求这N个人总共形成了多少社交网络
思路:
1.开一个数组course,用course[h]记录任意一个喜欢活动h的人的编号,findFather(course[h])就是这个人所在社交网络的根节点,于是对当前读入的人的编号i和他喜欢的每一个活动h,只需要合并i和findFather(course[h])即可。
2.开一个int型数组isRoot,isRoot[x]代表以x号人作为根节点的社交网络中有多少人,如果x不是根节点,则isRoot[x]为0,之后遍历i,让isRoot[findFather(i)]+1即可。
AC代码:
//1107 Social Clusters(30 分)
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 1010;
int father[maxn];
int isRoot[maxn] = { 0 };
int course[maxn] = { 0 };
int findFather(int x) {
int a = x;
while (x != father[x])x = father[x];
//路径压缩
while (a != father[a]) {
int z = a;
a = father[a];
father[z] = x;
}
return x;
}
void Union(int a, int b) {
int faA = findFather(a);
int faB = findFather(b);
if (faA != faB)father[faA] = faB;
}
void init(int n) {
for (int i = 1; i <= n; i++) {
father[i] = i;
isRoot[i] = false;
}
}
bool cmp(int a, int b) {
return a > b;
}
int main() {
int n, k, h;
int ans = 0;
(void)scanf("%d", &n);
init(n);
for (int i = 1; i <= n; i++) {
(void)scanf("%d:", &k);
for (int j = 0; j < k; j++) {
(void)scanf("%d", &h);
if (course[h] == 0)course[h] = i;//如果活动h第一次有人喜欢,令course[h]=i;
Union(i, findFather(course[h]));
}
}
for (int i = 1; i <= n; i++) {
isRoot[findFather(i)]++;
}
for (int i = 1; i <= n; i++) {
if (isRoot[i] != 0)ans++;
}
printf("%d\n", ans);
sort(isRoot + 1, isRoot + n + 1, cmp);
for (int i = 1; i <= ans; i++) {
printf("%d", isRoot[i]);
if (i < ans)printf(" ");
}
return 0;
}