【模板】并查集
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<iostream>
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;
}