并查集算法解决亲戚问题

题目：亲戚

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易。 现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。 我们规定：如果x和y是亲戚，y和z是亲戚，那么x和z也是亲戚；如果x,y是亲戚，那么x的亲戚都是y的亲戚，y的亲戚也都是x的亲戚。

输入

第一行：三个整数n,m,p，（n≤5000,m≤5000,p≤5000），分别表示有n个人，m个亲戚关系，询问p对亲戚关系。 以下m行：每行两个数Mi，Mj，1≤Mi，Mj≤N，表示Ai和Bi具有亲戚关系。 接下来p行：每行两个数Pi，Pj，询问Pi和Pj是否具有亲戚关系。

输出

P行，每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
样例输入

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

样例输出

Yes
Yes
No

#include<iostream>
using namespace std;
int *parent;//记录每个节点的指向
int *r;//rank数组 记录每个根的高度


int find(int x){
	while(x!=parent[x]){
		parent[x] = parent[parent[x]];//路径压缩
		x = parent[x];
	}
	return parent[x];
}

void unionElements(int x,int y){
	int xRoot = find(x);
	int yRoot = find(y);
	if(r[xRoot]<r[yRoot])		 
		parent[xRoot] = yRoot;//将矮的根指向高的根 降低树的深度
	else if(r[xRoot]>r[yRoot])
		parent[yRoot] = xRoot;
	else{
		parent[xRoot] = yRoot;
		r[yRoot]++;   //高度相同 随意指向  高度+1
	}
}

bool isConnected(int x,int y){
	return find(x) == find(y);
}

int main(){
	int n,m,p;
	cin>>n>>m>>p;
    r = new int[n+1];
    parent = new int[n+1];
	for(int i=1;i<=n;i++){
		r[i] = 1; //初始每个节点高度为1
		parent[i] = i;//节点指向自身
	}
	for(int i=0;i<m;i++){
		int x,y;
		cin>>x>>y;
		unionElements(x,y);
	}
	for(int i=0;i<p;i++){
		int x,y;
		cin>>x>>y;
		if(isConnected(x,y))
			cout << "Yes" << endl;
		else
			cout << "No" << endl;
	}
    system("pause");
    return 0;
}