蓝桥杯------蓝桥幼儿园

题目：

蓝桥幼儿园的学生是如此的天真无邪，以至于对他们来说，朋友的朋友就是自己的朋友。

小明是蓝桥幼儿园的老师，这天他决定为学生们举办一个交友活动，活动规则如下：

小明会用红绳连接两名学生，被连中的两个学生将成为朋友。

小明想让所有学生都互相成为朋友，但是蓝桥幼儿园的学生实在太多了，他无法用肉眼判断某两个学生是否为朋友。于是他请来了作为编程大师的你，请你帮忙写程序判断某两个学生是否为朋友（默认自己和自己也是朋友）。

输入描述
第 1 行包含两个正整数 N,M，其中 N表示蓝桥幼儿园的学生数量，学生的编号分别为  1∼N。

之后的第 2∼M+1 行每行输入三个整数，op , x , y

如果 op = 1，表示小明用红绳连接了学生 x和学生 y 。
如果 op=2，请你回答小明学生 x 和 学生 y 是否为朋友。
1≤N,M≤2*10^5 ,1≤x,y≤N。

输出描述
对于每个 op=2 的输入，如果 x和 y 是朋友，则输出一行 YES，否则输出一行 NO。

输入输出样例
示例 1

输入

5 5 
2 1 2
1 1 3
2 1 3
1 2 3 
2 1 2

输出

NO
YES
YES

分析:

看到朋友的朋友就是自己的朋友，我就已经知道这道题要用并查集来做。

操作1：连接x和y就是将x和y加入一个集合（merge）

操作2：判断x和y是否是朋友就是判断x和y是否在一个集合中   (find)

代码:

#include<iostream>
using namespace std;
const int MAX_N = 2*1e5+1;
int N,M;
int f[MAX_N];
int find(int x){
	if(x != f[x]){
		f[x] = find(f[x]);
	}
	return f[x];
}
int main(){
	cin >> N >> M;
	for(int i = 0;i < N;i++){
		f[i] = i;
	}
	int op,x,y;
	for(int i = 0;i < M;i++){
		cin >> op >> x >> y;
		if(op == 1){
			int fx = find(x);
			int fy = find(y);
			if(fx != fy){
				f[fx] = fy;
			}
		}else{
			if(find(x) == find(y)){
				cout << "YES";
			}else{
				cout << "NO";
			}
		}
	}
	return 0;
}