CD 66 并查集的实现

题目描述
给定一个没有重复值的整形数组arr，初始时认为arr中每一个数各自都是一个单独的集合。请设计一种叫UnionFind的结构，并提供以下两个操作。
boolean isSameSet(int a, int b): 查询a和b这两个数是否属于一个集合
void union(int a, int b): 把a所在的集合与b所在的集合合并在一起，原本两个集合各自的元素以后都算作同一个集合
[要求]
如果调用isSameSet和union的总次数逼近或超过O(N)，请做到单次调用isSameSet或union方法的平均时间复杂度为O(1)
输入描述:
第一行两个整数N, M。分别表示数组大小、操作次数
接下来M行，每行有一个整数opt
若opt = 1，后面有两个数x, y，表示查询(x, y)这两个数是否属于同一个集合
若opt = 2，后面有两个数x, y，表示把x, y所在的集合合并在一起
输出描述:
对于每个opt = 1的操作，若为真则输出"Yes"，否则输出"No"
示例1
输入
复制
4 5
1 1 2
2 2 3
2 1 3
1 1 1
1 2 3
输出
复制
No
Yes
Yes
说明
每次2操作后的集合为
({1}, {2}, {3}, {4})
({1}, {2, 3}, {4})
({1, 2, 3}, {4})

注意压缩路径

 #include<iostream>
 #include<stdio.h>
 #include<algorithm>
 using namespace std;
 
 int father[1000007];

int find(int x)
{
	if(x!=father[x])
	   father[x]=find(father[x]);//压缩路径，直接指向头节点 
	return father[x];
 
}
  
 int main()
 {
 	int n,m;
 
 	scanf("%d%d",&n,&m);
 	for(int i=1;i<=n;i++)
	 	father[i]=i;
	
	int k,x,y;
	for(int i=0;i<m;i++)
	{
		scanf("%d%d%d",&k,&x,&y);
		int x1,y1;
		x1=find(x);
		y1=find(y);
		
		if(k==1)
		{
			if( x1==y1  )
			   printf("Yes\n");
			else
			   printf("No\n");
		}
		else
		{
			father[x1]=y1;
		}		
	}
 	return 0;
 }