RQNOJ PID331 家族

题目描述

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

规定：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 n, m, k, pre[10010];
void init(){
    for(int i = 1; i <= n; i++){
        pre[i] = i;
    }
}
int getf(int v){
    if(pre[v] == v) return v;
    else{
        return pre[v] = getf(pre[v]);
    }
}
void meg(int v, int u){
    int t1 = getf(v), t2 = getf(u);
    if(t1 != t2){
        pre[t2] = t1;
    }
}
bool is_sam(int x, int y){
    int t1 = getf(x), t2 = getf(y);
    if(t1 == t2)    return true;
    else    return false;
}
int main(){
    cin >> n >> m >> k;
    init();
    for(int i = 1; i <= m; i++){
        int x, y;
        cin >> x >> y;
        meg(x, y);
    }
    for(int i = 1; i <= k; i++){
        int x, y;
        cin >> x >> y;
        if(is_sam(x, y))   cout << "Yes" << endl;
        else    cout << "No" << endl;
    }
    return 0;
}