洛谷-P1551 亲戚

原题链接：https://www.luogu.com.cn/problem/P1551

• 题目背景
• 题目描述
• 输入格式
• 输出格式
• 输入输出样例
• 说明/提示
• C++代码

---

题目背景

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

题目描述

规定：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

说明/提示

非常简单的并查集入门题哦！！！

C++代码

#include <iostream>
using namespace std;

int f[10005];

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

void combine(int x, int y) {
    f[find(y)] = find(x);
}

int main() {
    int n, m, p;
    cin >> n >> m >> p;
    int M[m][2], P[p][2];
    for (int i=0; i<m; ++i)
        cin >> M[i][0] >> M[i][1];
    for (int i=0; i<p; ++i)
        cin >> P[i][0] >> P[i][1];
    for (int i=1; i<=n; ++i)
        f[i] = i;
    for (int i=0; i<m; ++i)
        combine(M[i][0], M[i][1]);
    for (int i=0; i<p; ++i)
        if (find(P[i][0]) == find(P[i][1]))
            cout << "Yes" << endl;
        else
            cout << "No" << endl;
    return 0;
}