Disjointset 并查集（按秩合并，与路径压缩）的模板

并查集介绍 USACO 翻译：http://www.nocow.cn/index.php/%E5%B9%B6%E6%9F%A5%E9%9B%86

 

http://www.nocow.cn/index.php/Code:Disjointset_c%2B%2B

 

 

#include <iostream>
using namespace std;

int const MAXNUM = 500001;

int father[MAXNUM];

int rank[MAXNUM];
// 并查集 的三个函数initialSet, findSet, unionSet.
void initialSet(int i) //初始化并查集
{
    father[i] = i;
    rank[i] = 0;
}

int findSet(int i) //查找第i个 元素的集合号码, 回溯压缩路径. 
{
    if (i != father[i])
    {
        father[i] = findSet(father[i]);
    }
    return father[i];
}

void unionSet(int x, int y) // 合并两个集合, 通过rank[i] 判断元素加入集合的先后顺序. y 的rank 大于 x 的rank， x， y 均为集合号码。
{
    if (x == y)
        return;
    if (rank[x] > rank[y])
    {
        father[y] = x;
    }
    else
    {
        if (rank[x] == rank[y])
        {
            rank[y]++;
        }
        father[x] = y;
    }
}

 

 

 

另外一种并查集， 不用递归实现路径压缩。

#include <iostream> #include <vector> #include <utility> #include <algorithm> using namespace std; // 并查集的函数。 int n; int set[50010]; void initialSet(int n) { for (int i = 1; i <= n; i++) { set[i] = -1; } } int findSet(int i) { int topFather = i; int curFather = i; int temp; while (set[topFather] > 0) // 找到第i个元素所在集合的代表号码. { topFather = set[topFather]; } while (curFather != topFather) // 压缩路径. { temp = set[curFather]; set[curFather] = topFather; curFather = temp; } return topFather; } void unionSet(int x, int y) { int p1, p2; p1 = findSet(x); p2 = findSet(y); if (p1 == p2) { return; } else { if (set[p1] < set[p2]) { set[p1] = set[p1] + set[p2]; set[p2] = p1; } else { set[p2] = set[p1] + set[p2]; set[p1] = p2; } } } bool judge_connection(int t) // 判断如果割掉 t 点， 所有点是否在同一个集合。 { int p1, i; if (t == 1) { p1 = findSet(2); for (int i = 3; i <= n; i++) { if (findSet(i) != p1) { return false; } } return true; } else if(t == -1) { p1 = findSet(1); for (int i = 2; i <= n; i++) { if (findSet(i) != p1) { return false; } } return true; } else { p1 = findSet(1); for (int i = 2; i <= n; i++) { if (t == i) continue; if (findSet(i) != p1) { return false; } } return true; } }