注解
1、典型的并查集。注意初始化操作、合并操作、查找操作以及路径压缩操作。
2、这是并查集的模板题。给定的输入是两个朋友,将其合并成一个集合。最终输出的是集合数。
3、此题还可以扩展,当前是只查找到了集合的个数,如果要查找每个集合内的元素个数,把isRoot[findFather(i)] = 1;改成isRoot[findFather(i)]++;即可。然后遍历isRoot[]数组即可。
代码
#include <string.h>
#include <stdio.h>
const int MAX = 1001;
int N;
int father[MAX];
int isRoot[MAX];
void init() {
for(int i=0; i<=N; i++) {
father[i] = i;
}
}
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;
}
}
int main() {
int T;
scanf("%d", &T);
for(int i=0; i<T; i++) {
scanf("%d", &N);
init();
int m;
scanf("%d", &m);
for(int i=0; i<m; i++) {
int a, b;
scanf("%d %d", &a, &b);
Union(a, b);
}
memset(isRoot, 0, sizeof(isRoot));
for(int i=1; i<=N; i++) {
isRoot[findFather(i)] = 1;
}
int sum = 0;
for(int i=1; i<=N; i++) {
sum += isRoot[i];
}
printf("%d\n", sum);
}
return 0;
}