【并查集】HDU-1213 How Many Tables

注解

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

结果