洛谷P1551

洛谷P1551 亲戚 【并查集】

题目链接

题目背景

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

题目描述

规定：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，表示Mi和Mj具有亲戚关系。

接下来p行：每行两个数Pi，Pj，询问Pi和Pj是否具有亲戚关系。

输出格式

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

输入输出样例

输入 #1

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

输出 #1

Yes
Yes
No

解题思路

简单的并查集

Code

#include<bits/stdc++.h>

using namespace std;
const int maxn=5010;
int F[maxn],siz[maxn];

void Init_Set(){
    for(int i=0;i<maxn;i++){
        F[i]=i;
        siz[i]=0;
    }
}

int Find(int x){
    if(F[x]==x) return x;
    return F[x]=Find(F[x]);
}

void Union_Set(int x,int y){
    if((x=Find(x))==(y=Find(y))) return;
    if(siz[x]<siz[y]) F[x]=y;
    else if(siz[x]>siz[y]) F[y]=x;
    else{
        F[y]=x; siz[x]++;
    }
}

int main(){
    int n,m,p;
    cin>>n>>m>>p;
    Init_Set();
    int x,y;
    while(m--){
        cin>>x>>y;
        Union_Set(x,y);
    }
    while(p--){
        cin>>x>>y;
        if(Find(x)==Find(y)) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}