【模板】并查集
题目描述
输入输出格式
输入格式
接下来 M M M 行,每行包含三个整数 Z i , X i , Y i Z_i,X_i,Y_i Zi,Xi,Yi 。
当 Z i = 1 Z_i=1 Zi=1 时,将 X i X_i Xi 与 Y i Y_i Yi 所在的集合合并。
当
Z
i
=
2
Z_i=2
Zi=2 时,输出
X
i
X_i
Xi 与
Y
i
Y_i
Yi 是否在同一集合内,是的输出
Y
;否则输出 N
。
输出格式
输入输出样例
输入样例 #1
4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4
输出样例 #1
N
Y
N
Y
说明
对于 30 30% 30 的数据, N ≤ 10 N \le 10 N≤10, M ≤ 20 M \le 20 M≤20。
对于 70 % 70\% 70% 的数据, N ≤ 100 N \le 100 N≤100, M ≤ 1 0 3 M \le 10^3 M≤103。
对于 100 % 100\% 100% 的数据, 1 ≤ N ≤ 1 0 4 1\le N \le 10^4 1≤N≤104, 1 ≤ M ≤ 2 × 1 0 5 1\le M \le 2\times 10^5 1≤M≤2×105, 1 ≤ X i , Y i ≤ N 1 \le X_i, Y_i \le N 1≤Xi,Yi≤N, Z i ∈ { 1 , 2 } Z_i \in \{ 1, 2 \} Zi∈{1,2}。
#include<iostream>//并查集
using namespace std;
int parent[10002];
int find(int x){
while(parent[x]!=x){
x=parent[x]=parent[parent[x]];//路径压缩
}//循环找父节点,parent[i]等于i的时候就是最后的父节点
return x;
}
void root(int x,int y){
int x_root=find(x);
int y_root=find(y);
//要把两个点合并成一个就找到两个父节点,把其中一个父节点接过去
if(x_root!=y_root)parent[x_root]=y_root;//把x的父节点变成y的父节点,此时树就合并了
}//哪个作为哪个的父节点需要按情况而定,可能两个点代表不同的含义
int main()
{
int n,m;
int c,x,y;
cin>>n>>m;
for(int i=1;i<=n;i++){
parent[i]=i;//初始化,自己的父节点就是自己
}//初始化也可以每个设为-1,只要可以判断自己是父节点就可以啦
for(int i=1;i<=m;i++){
scanf("%d",&c);
if(c==1){
scanf("%d %d",&x,&y);
root(x,y);//合并操作
}
else{
scanf("%d %d",&x,&y);
if(find(x)==find(y))printf("Y\n");
else printf("N\n");//找根节点,如果根节点相等就在同一个集合
}
}
}
广东工业大学ACM寒假集训专题三A