PAT甲级 ——Birds in Forest

题目翻译
一些科学家为森林中成千上万的鸟类拍照。

假设所有出现在同一张照片中的鸟都属于同一棵树。

请你帮助科学家计算森林中树木的最大数量，对于任何一对鸟类，请判断它们是否在同一棵树上。

输入格式
第一行包含整数 N 表示照片数量。
接下来 N 行，每行描述一张照片，格式如下：K B1 B2 … BK，K 表示照片中的鸟的数量，B_i 是鸟的具体编号。

保证所有照片中的鸟被连续编号为 1 到某个不超过 10⁴ 的整数。

再一行包含整数 Q。
接下来 Q 行，每行包含两个鸟的编号，表示一组询问。

输出格式
第一行输出最大可能的树的数量以及鸟的数量。
接下来对于每个询问，如果被询问的两个鸟在同一棵树上，则在一行中输出 Yes，否则输出 No。

数据范围
N ≤ 10⁴
Q ≤ 10⁴

输入样例
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

---

题解
并查集：

解题思路：由于鸟的编号连续，所以鸟的数量就是所出现编号的最大值；

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

const int N = 10010;

int n, m;
int k, a, b;
int p[N];

int find(int x)
{
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main()
{
    cin >> n;
    
    for (int i = 1; i <= N; i ++) p[i] = i;
    
    int Num = 0;
    while(n --)
    {
        cin >> k >> a;
        Num = max(Num, a);
        for (int i = 1; i < k; i ++)
        {
            cin >> b;
            Num = max(Num, b);
            int pa = find(a), pb = find(b);
            p[pb] = pa;
        }
    }
    
    int cnt = 0;
    for (int i = 1; i <= Num; i ++)
        if(find(i) == i) cnt ++;
        
    cout << cnt << " " << Num << endl;
    
    cin >> m;
    while(m --)
    {
        cin >> a >> b;
        if(find(a) == find(b)) puts("Yes");
        else puts("No");
    }
    
    return 0;
}

ps：发现了一个神奇的问题，把第 9 行的代码放到第 6 行后，在本地编译器上就只会输出 2 10，但是在 PAT 上还是能 AC😱