2021-04-13-做题- 并查集-P1551 亲戚

1. 描述

洛谷 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是否具有亲戚关系。

2. 分析

算法学习笔记(1) : 并查集

3. 代码

#include <iostream>
#define MAXN 5005

using namespace std;

int fa[MAXN];
int dep[MAXN];

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

int find(int x)
{
    return x == fa[x] ? x : (fa[x] = find(fa[x]));
}

inline void merge(int i, int j)
{
    int x = find(i), y = find(j);
    if (dep[x] <= dep[y])
        fa[x] = y;
    else
        fa[y] = x;
    if (dep[x] == dep[y] && x != y)
        dep[y]++;
}

int main()
{
    int n, m, p, x, y;
    cin >> n >> m >> p;

    init(n);

    for (int i = 0; i < m; ++i)
    {
        cin >> x >> y;
        merge(x, y);
    }

    for (int i = 0; i < p; ++i)
    {
        cin >> x >> y;
        if(find(x) == find(y)){
            cout << "Yes" << endl;
        }
        else{
            cout << "No" << endl;
        }
    }
    return 0;
}

结果

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
Yes
2 3
Yes
5 6
No