PAT(Advanced)1118 Birds in Forest C++实现 并查集 路径压缩

PAT(Advanced)甲级1118 Birds in Forest C++实现 并查集 路径压缩

题目链接

1118 Birds in Forest

题目大意

给定N张照片，N张照片中有M只鸟，题目认为出现在同一张照片中的鸟属于同一棵树，求解N张照片中存在几棵树、几只鸟，鸟的编号是从1到某个值，对于Q次询问，每次询问为两只鸟的序号，若两只鸟出现在同一棵树上，输出Yes，否则输出No

算法思路

根据题意，若两张照片中出现同一只鸟则说明两张照片的所有鸟在一棵树上，依次类推，利用带路径压缩的并查集，对于每张照片出现的鸟，若指向的根节点不同，则进行合并。依题意，求出鸟的最大序号即为鸟的数量，根据并查集性质可以求得树的数量，即求解图中存在几个连通分量，对于每次询问，查找两只鸟的根节点，判断是否在同一个连通分量中

AC代码

/*
author : eclipse
email  : eclipsecs@qq.com
time   : Fri Jun 26 10:24:23 2020
*/
#include <bits/stdc++.h>
using namespace std;

const int MAX_SIZE = 10e4 + 10;
vector<int> roots;

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

int unionRoot(int x, int y) {
    int rootX = find(x);
    int rootY = find(y);
    roots[rootY] = rootX;
}

int main(int argc, char const *argv[]) {
    int N;
    roots.resize(MAX_SIZE);
    int birdsNumber = 0;
    scanf("%d", &N);
    for (int i = 0; i < N; i++) {
        int K;
        scanf("%d", &K);
        int root;
        for (int j = 0; j < K; j++) {
            int bird;
            scanf("%d", &bird);
            birdsNumber = max(birdsNumber, bird);
            if (!roots[bird]) {
                roots[bird] = bird;
            }
            if (j == 0) {
                root = bird;
            } else {
                unionRoot(root, bird);
            }
        }
    }

    int treeNumber = 0;
    for (int i = 1; i <= birdsNumber; i++) {
        treeNumber += roots[i] == i ? 1 : 0;
    }
    printf("%d %d\n", treeNumber, birdsNumber);
    int Q;
    scanf("%d", &Q);
    for (int i = 0; i < Q; i++) {
        int bird1, bird2;
        scanf("%d%d", &bird1, &bird2);
        printf("%s\n", find(bird1) == find(bird2) ? "Yes" : "No");
    }
    return 0;
}

样例输入

4
3 10 1 2
2 3 4
4 1 5 7 8
3 9 6 4
2
10 5
3 7

样例输出

2 10
Yes
No

鸣谢

PAT

最后

• 由于博主水平有限，不免有疏漏之处，欢迎读者随时批评指正，以免造成不必要的误解！