【模板】并查集

【模板】并查集


from luogu P3367
时间限制 :1s
内存限制:125M

题目描述:

如题,现在有一个并查集,你需要完成合并和查询操作。

输入格式:

第一行包含两个整数 N,M ,表示共有 N 个元素和 M 个操作。

接下来 M 行,每行包含三个整数 Zi,Xi,Yi

当 Zi=1 时,将 Xi 与 Yi 所在的集合合并。

当 Zi=2时,输出 Xi 与 Yi 是否在同一集合内,是的输出 Y ;否则输出 N

输出格式:

对于每一个 Zi=2 的操作,都有一行输出,每行包含一个大写字母,为 Y 或者 N 。

样例输入:

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

样例输出:

N
Y
N
Y

说明/提示:

对于 100% 的数据,0 < N <= 10 ^ 4,1 0 < M <= 2 * 10 ^ 5。


这是一道并查集很基础的题目,就直接上代码吧!

AC代码:

#include
using namespace std;
int father[10005];          //记录每个节点的所在树的代表节点,也可以理解为该点的父节点
int z,x,y,fx,fy;
int find(int &x){
     
    if(father[x] == x)
        return x;
    return father[x] = find(father[x]);
}
int main() {
     
    int n, m;
    cin>>n>>m;
    for(int i = 1;i <= n;++i) father[i] = i;    //初始化每个点的父亲为其本身
    while(m--) {
     
        cin >> z >> x >> y;
        fx = find(x), fy = find(y);        //找到x和y的father
        if (z == 1) {
     
            if (fx != fy)
                father[fx] = fy;           //如果z=1那么合并x和y所在的树
        } else {
     
            if (fx == fy) cout << "Y\n";
            else cout << "N\n";				//否则输出x和y是否在同一棵树
        }
    }
	return 0;
}

你可能感兴趣的:(并查集,最小生成树,数据结构,算法,c++)