【题解】【AcWing】1608. 森林里的鸟

1608. 森林里的鸟

原题传送：AcWing 1608. 森林里的鸟

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

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

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

输入格式

第一行包含整数 $N$ 表示照片数量。

接下来 $N$ 行，每行描述一张照片，格式如下：

$K$ $B_1$ $B_2$ … $B_K$

$K$ 表示照片中的鸟的数量， $B_i$ 是鸟的具体编号。

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

再一行包含整数 $Q$ 。

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

输出格式

第一行输出最大可能的树的数量以及鸟的数量。

接下来对于每个询问，如果被询问的两个鸟在同一棵树上，则在一行中输出 Yes，否则输出 No。

数据范围

$1\le N\le 10^4$ ,
$1\le K\le 10$ ,
$1\le Q\le 10^4$

输入样例：

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 <bits/stdc++.h>

using namespace std;

const int N = 10010;

int n;
int birds[10];
int p[N];
bool st[N];

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

int main()
{
	scanf("%d", &n);
	
	for(int i = 1; i <= N; i++)
		p[i] = i;
		
	int cnt = 0;
	for(int i = 1; i <= n; i++)
	{
		int k;
		scanf("%d", &k);
		
		for(int j = 0; j < k; j++)
		{
			scanf("%d", &birds[j]);
			st[birds[j]] = true;
		}
		
		for(int j = 1; j < k; j++)
		{
			int a = birds[j - 1], b = birds[j];
			a = find(a), b = find(b);
			if(a != b)
			{
				p[a] = b;
				cnt++;
			}
		}
	}
		
	int tot = 0;
	for(int i = 0; i < N; i++)
		tot += st[i];
	printf("%d %d\n", tot - cnt, tot);
	
	int q;
	scanf("%d", &q);
	while(q--)
	{
		int a, b;
		scanf("%d%d", &a, &b);
		if(find(a) == find(b))
			printf("Yes\n");
		else
			printf("No\n");
	}
	
	return 0;
}