洛谷P1551 亲戚 题解 并查集入门题

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

解题思路

这道题是最基础的并查集入门题。

我们甚至可以不适用并查集的路径优化。

暴力的查找祖先节点的方式如下：

int func_find(int x) {
    while (f[x] != x)
        x = f[x];
}

添加路径压缩后的查找函数如下：

int func_find(int x) {  // 暴力查找
    int y = x;
    while (f[x] != x)
        x = f[x];
    while (f[y] != x)
        f[y] = x;
}

递归写法如下：

int func_find(int x) {
    return x == f[x] ? x : f[x] = func_find(f[x]);
}

完整实现代码如下：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5050;
int n, m, p, f[maxn];
int func_find(int x) {
    return x == f[x] ? x : f[x] = func_find(f[x]);
}
int func_union(int x, int y) {
    int a = func_find(x), b = func_find(y);
    f[a] = f[b] = f[x] = f[y] = min(a, b);
}
void init() {
    for (int i = 1; i <= n; i ++) f[i] = i;
}
int main() {
    cin >> n >> m >> p;
    init();
    while (m --) {
        int x, y;
        cin >> x >> y;
        func_union(x, y);
    }
    while (p --) {
        int x, y;
        cin >> x >> y;
        puts( func_find(x) == func_find(y) ? "Yes" : "No" );
    }
    return 0;
}