并查集模板

并查集模板题目: P3367

普通并查集模板：

初始化：

并查集数组每个元素初始自成一个集合，所以每个元素的值为其本身。所处的树的深度初始化为1

 

void init(int n)
{
    for(int i = 1; i <= n; i++)
    {
        f[i] = i;
        r[i] = 1;
    }
}

 

代表元素的查找：

带有路径压缩的查找，递归写法：

 

int getfather(int x)
{
    return f[x] == x?x:(f[x] = getfather(f[x]));
}

 

注意带括号，三元运算符优先级比等号高

两个集合的合并：

所处深度小的合并到深度大的

 

void mergeset(int x, int y)
{
    x = getfather(x); y = getfather(y); // 注意是父节点的合并
    if(getfather(x) != getfather(y))
    {
        if(r[x] <= r[y]) f[x] = y;
        else f[y] = x;
        if(r[x] == r[y] && x != y) r[x]++;
    }
}

 

整体代码（根据模板题写的）：

 

#include <iostream>
#include <cstdio>
#define N 200010
using namespace std;
int f[N],r[N];
void init(int n)
{
    for(int i = 1; i <= n; i++)
    {
        f[i] = i;
        r[i] = 1;
    }
}
int getfather(int x)
{
    return f[x] == x?x:(f[x] = getfather(f[x]));
}
void mergeset(int x, int y)
{
    x = getfather(x); y = getfather(y); 
    if(getfather(x) != getfather(y))
    {
        if(r[x] <= r[y]) f[x] = y;
        else f[y] = x;
        if(r[x] == r[y] && x != y) r[x]++;
    }
}
int main()
{
    int n,m,z,x,y;
    cin >> n >> m;
    init(n);
    while(m--)
    {
        cin >> z >> x >> y;
        if(z == 1) mergeset(x,y);
        else if(z == 2) cout << (getfather(x) == getfather(y)?'Y':'N') << endl;
    }
    return 0;
}