PAT--1118. Birds in Forest（并查集）

最新推荐文章于 2023-11-08 14:54:35 发布

原创最新推荐文章于 2023-11-08 14:54:35 发布 · 918 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pat #并查集

PAT 同时被 3 个专栏收录

96 篇文章

订阅专栏

数据结构

26 篇文章

订阅专栏

数据结构--并查集

8 篇文章

订阅专栏

博客介绍了PAT考试中的一道题目1118——Birds in Forest，主要讨论了如何运用并查集解决这个问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1118. Birds in Forest

题解

裸的并查集

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 10000 + 5;
const int inf  = 0x3f3f3f3f;
int p[maxn], tmp[maxn];
int n, k, q;

int find(int x) { return p[x] == x ? x : p[x] = find(p[x]); }
void Union(int u, int v){
    int pu = find(u), pv = find(v);
    if(pu != pv) p[pu] = pv;
}

int main(){
#ifdef EXMY
freopen("data.in", "r", stdin);
#endif // EXMY
    for(int i = 1; i < maxn; ++i) p[i] = i;
    int tot = 0;
    int Max = -inf;
    cin >> n;
    while(n--){
        cin >> k;

        int t, r;
        cin >> t;
        Max = max(Max, t);
        for(int i = 1; i < k; ++i){
            cin >> r;
            Max = max(Max, r);
            Union(t, r);
            t = r;
        }
    }
    for(int i = 1; i <= Max; ++i){
        if(p[i] == i) tot++;
    }
    cout << tot << " " << Max << endl;


    cin >> q;
    int u, v;
    while(q--){
        scanf("%d %d", &u, &v);
        if(find(u) == find(v)) printf("Yes\n");
        else printf("No\n");
    }

    return 0;
}