【并查集】P1551 亲戚（洛谷）

题目背景

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

题目描述

规定：xx 和 yy 是亲戚，yy 和 zz 是亲戚，那么 xx 和 zz 也是亲戚。如果 xx，yy 是亲戚，那么 xx 的亲戚都是 yy 的亲戚，yy 的亲戚也都是 xx 的亲戚。

输入格式

第一行：三个整数 n,m,pn,m,p，（n,m,p \le 5000n,m,p≤5000），分别表示有 nn 个人，mm 个亲戚关系，询问 pp 对亲戚关系。

输出格式

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

输入输出样例

输入

输出

#include <iostream>
using namespace std;
int p[10005];


void init(int n) {
    for(int i = 1; i <= n; i++)  p[i] = i;
}

int find(int i) {
    return i == p[i] ? i : (find(p[i]));
}

void merge(int i, int j) {
    p[find(i)] = find(j);
}

int main()
{
    int n, m, p;
    cin>>n>>m>>p;
    init(n);
    for(int i = 0;i < m; i++) {
        int a, b;
        cin>>a>>b;
        merge(a, b);
    }
    for(int i = 0;i < p; i++) {
        int a, b;
        cin>>a>>b;
        cout<<(find(a) == find(b) ? "Yes" : "No") <<endl;
    }
    return 0;
}