【codevs 1073】家族

题目描述 Description

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

输入描述 Input Description

第一行：三个整数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是否具有亲戚关系。

输出描述 Output Description

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

样例输入 Sample Input

6 5 3

1 2

1 5

3 4

5 2

1 3

1 4

2 3

5 6

样例输出 Sample Output

Yes

Yes

No

数据范围及提示 Data Size & Hint

n<=5000,m<=5000,p<=5000

#include<cstdio>
#include<iostream>
#define maxn 5005
using namespace std;
int n,m,p,a,b,fa[maxn],deep[maxn];
int read(){ 
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void init(int n){
    for(int i=0;i<n;i++){
        fa[i]=i;
        deep[i]=0;
    }
}
int find(int x){
    if(x==fa[x]) return x;
    else return fa[x]=find(fa[x]);
}
void unite(int x,int y){
    x=find(x),y=find(y);
    if(x==y) return ;
    if(deep[x]<deep[y]) fa[x]=y;
    else{
        fa[y]=x;
        if(deep[x]==deep[y]) deep[x]++;
    }
}
bool same(int x,int y){
    return find(x)==find(y);
}
int main(){
    n=read(),m=read(),p=read();
    init(n);
    for(int i=1;i<=m;i++){
        cin>>a>>b;
        unite(a,b);
    }
    for(int i=1;i<=p;i++){
        cin>>a>>b;
        if(same(a,b)) printf("Yes
");
        else printf("No
"); 
    }
    return 0;
}