《算法笔记》9.6 并查集

一.定义
实现：int father[N];
father[i] = j; 表示 j 的父亲结点是 i；father[i] = i; 表示元素 i 是该集合的根结点；

二.基本操作
1.初始化

for(int i = 1; i <= n; i++) father[i] = i;

2.查找
非递归实现：

int findFather(int x) {
	while(x != father[x]) x = father[x];
	return x;
}

递归实现：

int findFather(int x) {
	if(x == father[x]) return x;
	else return findFather(father[x]);
}

3.合并
给出元素 a、b，把两个元素所在的集合合并，思路：把其中一个集合的根结点的父亲指向另一个集合的根结点；（合并产生的每一个集合都是一棵树）

void Union(int a, int b) {
	int faA = findFather(a);
	int faB = findFather(b);
	if(faA != faB) father[faA] = faB;
}

三.路径压缩
把当前查询结点的路径上的所有结点的父亲都指向根结点，查询复杂度可以降为 O(1)；
1.非递归写法：

int findFather(int x) {
	int a = x;
	while(x != father[x]) x = father[x]; 
	// 现在 x 存放的是根结点，下面把所有结点的 father 都改成根结点
	while(a != father[a]) {
		int z = a;
		a = father[a];
		father[z] = x;
	} 
	return x;
}

2.递归写法：

int findFather(int v) {
	if(v == father[v]) return v;
	else {
		int F = findFather(father[v]); //最后找到的 F 是 v 所在集合的根结点 
		father[v] = F; //把路径上每个结点的 father 都改成根结点 
		return F; //返回 v 的根结点 
	}
}

四.题目
1.好朋友
思路：

这种关系就是并查集的关系，一个好朋友集合就是一个并查集，用 Union函数构造并查集；
计算并查集数：每个结点都找到根结点，利用一个 bool 数组标记；

注意：
标号从1开始；

代码：

#include<cstdio>
using namespace std;
const int maxn = 110;
int n, m;
int fred[maxn];
bool flag[maxn];

void init() {
	for(int i = 1; i <= n; i++) {
		fred[i] = i;
		flag[i] = false;
	}
}

int findFriend(int a) {
	while(a != fred[a]) a = fred[a];
	return a;
}

void Union(int a, int b) {
	int faA = findFriend(a);
	int faB = findFriend(b);
	if(faA != faB) fred[faB] = faA;
}

int main() {
	scanf("%d%d", &n, &m);
	init();
	int f1, f2;
	for(int i = 0; i < m; i++) {
		scanf("%d%d", &f1, &f2);
		Union(f1, f2);
	}
	for(int i = 1; i <= n; i++) flag[findFriend(i)] = true;
	int ans = 0;
	for(int i = 1; i <= n; i++) {
		if(flag[i]) ans++;
	}
	printf("%d", ans);
	return 0;
} 

2.PAT A1107
思路：
如何合并：合并的条件是有相同 hobby，所以可以列出一个数组 course[hobby] ，内容是有此 hobby 的人的编号。每次读入某人的 hobby 时，将此人与 course[hobby] 的 father 合并（father 是第一个有此 hobby 的人）；
如何计算共有多少并查集+每个并查集的人数：

for(int i = 1; i <= n; i++) {
		isRoot[findFather(i)]++;
	}
	for(int i = 1; i <= n; i++) {
		if(isRoot[i] != 0) ans++;
	}

代码：

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1010;
int n, ans = 0;
int course[maxn] = {0};
int isRoot[maxn] = {0};
int father[maxn];

void init() {
	for(int i = 1; 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[faB] = faA; 
}

bool cmp(int a, int b) {
	return a > b;
}

int main() {
	scanf("%d", &n);
	init();
	int num, hobby;
	for(int i = 1; i <= n; i++) {
		scanf("%d:", &num);
		for(int j = 0; j < num; j++) {
			scanf("%d", &hobby);
			if(course[hobby] == 0) course[hobby] = i;
			Union(i, findFather(course[hobby]));
		}
	}
	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;
}