亲属关系--并查集训练T1

题目描述
T1.亲属关系(relation.pas)

【问题描述】

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。

规定：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>
#include <algorithm>
#include <vector>
#include <string>
#include <math.h>
using namespace std;
const int maxn=5005;
int n,m,p;
int a[maxn];
int find(int x){
    if(x==a[x]){
        return x;
    }
    return a[x]=find(a[x]);
}
void merge(int x,int y){
    int xf=find(x);
    int yf=find(y);
    if(xf!=yf){
        a[xf]=yf;
    }
}
int main(){
    while(cin>>n>>m>>p){
        int x,y;
        for(int i=1;i<maxn;i++){
            a[i]=i;
        }
        for(int i=1;i<=m;i++){
            cin>>x>>y;
            merge(x,y);
        }
        for(int i=1;i<=p;i++){
            int a,b;
            cin>>a>>b;
            int af=find(a);
            int bf=find(b);
            if(af!=bf){
                cout<<"No"<<endl;
            }else{
                cout<<"Yes"<<endl;
            }
        }    
    }
    return 0;
}