1118 Birds in Forest

题目大意：

有n张照片，每张上面的鸟来自同一棵树，问有多少棵树和鸟，并且查询一对鸟是否来自同一棵树。

解题思路：

并查集板子题，合并每一对输入的鸟，用数组记录鸟的编号并统计鸟的数量，最后用father数组统计树的数量。
代码如下：

#include<iostream>
#include<cstdio>
#include<set>
#include<algorithm>
using namespace std;
int father[11000],book[11000],n,q;
int getf(int f)//一开始用递归写法有一个3分测试点超时了 
{
	int a=f;
	while(f!=father[f])
	{
		f=father[f];
	}
	while(a!=father[a])
	{
		int z=a;
		a=father[a];
		father[z]=f;
	}
	return f;
}
void merge(int a,int b)
{
  int fa=getf(a);
  int fb=getf(b);
  if(fa!=fb)
  {
  	father[fb]=fa;
  }	
}
void init()//预处理father数组 
{
	for(int i=1;i<=10000;i++)
	 father[i]=i;
}
int main()
{
	init();
	scanf("%d",&n);
	int num1=0,num2=0;
	for(int i=1;i<=n;i++)
	{
		int t,tmp[1100]={0};
		scanf("%d",&t);
		for(int j=0;j<t;j++)
		{
			scanf("%d",&tmp[j]);
			if(book[tmp[j]]==0)
			{
				num2++;//鸟的数量 
				book[tmp[j]]=1;
			}
			if(j!=0)
				merge(tmp[j-1],tmp[j]);
		 } 
	}
	for(int i=1;i<=num2;i++)//统计树的数量 
	{
		if(book[i]&&father[i]==i)num1++;
	}
	scanf("%d",&q);
	printf("%d %d\n",num1,num2);
	while(q--)
	{
		int a,b;
		scanf("%d %d",&a,&b);
		if(getf(a)==getf(b))printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}