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

简单的并查集

#include<cstdio>
#include<iostream>
#include<vector>
#include<iomanip>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iterator>
using namespace std;

int n, m, p;
int f[5005];

void init() {
    for (int i = 0; i < 5005; i++)f[i] = i;
}


int find(int a) {
    if (a == f[a])return a;
    else return f[a] = find(f[a]);
}

void merge(int a, int b) {
    a = find(a);
    b = find(b);
    if (a != b)f[a] = b;
}

int main()
{
    int mi, mj;
    cin >> n >> m >> p;
    init();
    for (int i = 1; i <= m; i++) {
        cin >> mi >> mj;
        merge(mi, mj);
    }
    int pi;
    int pj;
    for (int i = 1; i <= p; i++) {
        cin >> pi >> pj;
        int a = find(pi);
        int b = find(pj);
        
        if (a == b)cout << "Yes" << endl;
        else cout << "No" << endl;
    }

    return 0;
}