【洛谷】P3367 【模板】并查集 题解

原题地址：https://www.luogu.org/problem/P3367

题目描述

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

输入输出格式

输入格式：

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

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

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

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

输出格式：

如上，对于每一个Zi=2的操作，都有一行输出，每行包含一个大写字母，为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

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据，N<=10，M<=20；

对于70%的数据，N<=100，M<=1000；

对于100%的数据，N<=10000，M<=200000。

思路：并查集模板题。

并查集，在有N个元素的集合中，通常是在开始时让每个元素构成一个单元素的集合（即将i最高领导初始化为自己），然后按一定顺序将属于同一组的元素所在的集合合并（属于同一集合的元素合并在一起），其间要反复查找一个元素在哪个集合中。

推荐一篇博客：并查集


代码如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=1e6;
int n,m,zi,xi,yi;
int f[N];	//表示一个元素的上级 
int find(int a){	//找到自己的最高领导 
	if(f[a]==a)		//如果自己的最高领导是自己，返回自己 
		return a;
	return f[a]=find(f[a]);	//否则继续寻找，直到找到 
}
int main(){
	ios::sync_with_stdio(false);	//取消同步 
	cin>>n>>m;
	for(int i=1;i<=m;i++)
		f[i]=i;		//初始化i的最高领导是自己 
	for(int i=1;i<=m;i++){
		cin>>zi>>xi>>yi;
		if(zi==1){
			f[find(xi)]=find(yi);	//合并，使xi和yi在同一集合 
		} 
		else{
			if(find(xi)==find(yi))	//如果xi和yi在同一集合 
				cout<<"Y"<<endl;
			else
				cout<<"N"<<endl;
		}
	}
	return 0;
}