核心代码(C++版本)
const int MAXNUM = 500000 + 10;
int id[MAXNUM];
int Size[MAXNUM];
int Count;
// 初始化
void make_set(int n){
for(int i = 0 ; i < n; i++){
id[i] = i;
Size[i] = 1;
}
Count = n;
}
// 查找父节点
int Find(int p) {
while (p != id[p]) {
// 路径压缩,会破坏掉当前节点的父节点的尺寸信息,因为压缩后,当前节点的父节点已经变了
id[p] = id[id[p]];
p = id[p];
}
return p;
}
// 合并 p ,q节点
void Union(int p, int q) {
int pRoot = Find(p);
int qRoot = Find(q);
if (pRoot == qRoot) {
vaild = false;
return;
}
// 按秩进行合并
if (Size[pRoot] > Size[qRoot]) {
id[qRoot] = pRoot;
Size[pRoot] += Size[qRoot];
} else {
id[pRoot] = qRoot;
Size[qRoot] += Size[pRoot];
}
// 每次合并之后,树的数量减1
Count--;
}
// 森林中树的数量
int TreeNumber() {
return Count;
}
HDU 1213 : How Many Tables
模版题,题意是问Ignatius生日派对上需要准备多少张桌子,规则是相互认识的朋友坐在一桌上,一共n个人,有m种关系。
也就是求出并查集中的树的个数。
HDU 1232 : 畅通工程
由题意是求至少需要加上多少条边使得每个城市都能与其他城市相通(至少有一条路径到达)。在已知的道路中可以形成并查集里的树,每棵树内每个城市都至少有一条道路能到达其他城市。此时可以看出所求的边数就是森林中树的个数减去1,也就是Count-1。
POJ 2524 : Ubiquitous Religions
题目求大学里有多少不同宗教的学生,每个学生最多只能申请一种宗教。许多学生不愿意表达自己的信念。 避免这些问题的一种方法是询问m(0 <= m <= n(n-1)/ 2)对学生,并询问他们是否相信同一个宗教。这里就是求并查集中树的个数,其中【询问他们是否相信同一个宗教,而不是问具体哪个宗教】,也就是注重点在于哪颗树,而不是树的序列号,这样我们直接计算树的数量即可。
POJ 1611 : The Suspects
题意:题意假设0号学生是传染非典的人,然后给出m组人的情况,在同一组都会被认为是传染非典的人,求传染非典的人数。实际上就是将同一个组通过并查集的【并】操作合并在一起,然后计算出和0号学生同一组的人数,也就是Size[Find(0)]的值。
POJ 2236 : Wireless Network
网络中的所有计算机全部断开。
(1)由于硬件限制,每台计算机只能直接与距离它不远的计算机进行通信
(2)如果计算机A和计算机B可以直接通信,或者计算机C可以同时与A和B通信B
O操作是修复第x号计算机,
S操作是测试x,y是否能够通信
解答:保存每个计算机的坐标,对于O操作,表示修复X节点,则判断X和之前修复的节点的距离是否小于等于d,如果是则连接X和该节点,其中已经修复的节点使用map保存,然后之后遍历map即可。对于S操作,直接判断Find(X)==Find(Y).
LeetCode 128 : Longest Consecutive Sequence
求最长连续序列的长度。
Input: [100, 4, 200, 1, 3, 2]
Output: 4
Explanation: The longest consecutive elements sequence is [1, 2, 3, 4]. Therefore its length is 4.
在序列a[n]中,对于每一个a[i],检查a[i]-1和a[i]+1,这里使用map直接对应,如果存在则通过并查集的Union操作合并在一起。最终检查Size数组的最大值,或者通过中间变量Max时刻记录。
for(int i = 0 ; i < n ; i ++){
scanf("%d", &a[i]);
// 标记
mp[a[i]] = 1;
if(mp.find(a[i] + 1)!= mp.end())
Union(a[i], a[i] +