并查集的操作:
1. 将两个集合合并
2. 询问两个元素是否在一个集合当中
基本原理:每个集合以一颗树来表示,树的根节点即为整个集合的编号。每个节点存储它的父节点。
p[x] 表示x 的父节点
问题1:如何判断树根: if(p[x] == x)
问题2:如何求x集合的编号: while(x != p[x]) x = px[x];
问题3: 如何合并两个集合:px 是x的集合编号,py是y的集合编号。p[x] = y;
例题:
题目来源:洛谷 / 题目列表 / 题目详情 P3367 【模板】并查集
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数 N,MN,M ,表示共有 NN 个元素和 MM 个操作。
接下来 MM 行,每行包含三个整数 Z_i,X_i,Y_iZi,Xi,Yi 。
当 Z_i=1Zi=1 时,将 X_iXi 与 Y_iYi 所在的集合合并。
当 Z_i=2Zi=2 时,输出 X_iXi 与 Y_iYi 是否在同一集合内,是的输出 Y
;否则输出 N
。
输出格式
对于每一个 Z_i=2Zi=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
#include<iostream>
#include<cstdio>
using namespace std;
const int N = 1e5;
int p[N];
int n, m;
int find(int x)
{
if(x != p[x]) p[x] = find(p[x]);
return p[x];
}
int main()
{
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++) p[i] = i;
int a, b, c;
while(m--)
{
scanf("%d %d %d", &a, &b, &c);
if(a == 1) p[find(b)] = find(c);
else
{
if(find(b) == find(c)) printf("Y\n");
else printf("N\n");
}
}
return 0;
}