1118 Birds in Forest (25 分)

原题目： 1118 Birds in Forest (25 分).

题意

假设一张照片中的鸟儿属于同一棵树。
给出照片数量N，接下来N行，给出每张照片的小鸟数量K以及小鸟的ID（1~10⁴），然后，给出正数Q以及Q对鸟儿的ID。
① 输出树的最多可能数量和鸟儿的数量，
② 判断给出的对子是否属于同一棵树 。

分析

1. 题意分析：同一只鸟可能出现在不同照片中，而同一张照片中鸟儿都属于同一棵树，需要计算树的数量和鸟儿的数量，因此，需要将同属一棵树但在不同照片中的鸟儿合并到一起——并查集问题——先写好三个基本操作（这题需要在findFather中优化路径，否则有一个测试点会超时）。
2. 鸟儿的数量可以用一个visit数组标记是否在照片中出现过来计算（静态数组的随机访问性）。
3. 树的数量：① 可以用集合set无重复元素的特性，通过调用size()得到；② 也可以用一个cnt数组累计根结点为findFather的树的数量，然后统计非0元素的个数。
4. 判断两只鸟是否属于同一棵树，只需要知道根结点是否相同即可。

知识点

1. 并查集的应用。
2. 静态数组的随机访问性。
3. 利用集合set无重复元素的特性利用.size()进行数量统计。

CODE

#include <iostream>
#include <set>
using namespace std;
const int maxn = 10005;
int father[maxn];
bool visit[maxn];  //记录鸟儿是否出现
int cnt = 0;  //用于存储鸟儿的数量
set<int> s;  //用于计算树的数量

int findFather(int x);
void Union(int a, int b);

int main()
{
    int n, k;
    //初始化并查集
    for ( int i=1; i<=maxn; i++ )
        father[i] = i;
    //输入并合并
    cin >> n;
    for ( int i=0; i<n; i++ ){
        int id;
        cin >> k >> id;  //先输入这张照片中的第一只鸟儿的ID
        visit[id] = true;  //标记为出现过
        for ( int j=0; j<k-1; j++ ){  //注意：这里只剩k-1只鸟儿需要输入
            int tmp;
            cin >> tmp;
            visit[tmp] = true;  //小鸟id在照片中出现
            Union(id, tmp);  //合并
        }
    }
    for ( int i=1; i<=maxn; i++ ){  //遍历所有小鸟
        if ( visit[i] ){
            int fa = findFather(i);
            cnt++;  //鸟儿的总数
            s.insert(fa);  //树的数量
        }
    }
    printf("%d %d\n", s.size(), cnt);
    //判断是否同一棵树
    int q;
    cin >> q;
    for ( int i=0; i<q; i++ ){
        int a, b;
        cin >> a >> b;
        int f1 = findFather(a);
        int f2 = findFather(b);
        if ( f1==f2 ) cout << "Yes" << endl;
        else cout << "No" << endl;
        //更简洁的写法
        //printf("%s\n", (findFather(a)==findFather(b)) ? "Yes" : "No");
    }
    
    return 0;
}
int findFather(int x){
    int a = x;
    while( x!=father[x] )
        x = father[x];
    while( a!=father[a] ){
        int z = a;
        a = father[a];
        father[z] = x;
    }
    return x;
}
void Union(int a, int b){
    int faA = findFather(a);
    int faB = findFather(b);
    if ( faA!=faB )
        father[faA] = faB;
}