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  : [email protected]
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

最後

• 由於博主水平有限，不免有疏漏之處，歡迎讀者隨時批評指正，以免造成不必要的誤解！