PTA 看照片找基友 (20分)

PTA 看照片找基友 (20分)

题目：

小A是个单身狗，他有很多好基友，他们平时喜欢出去聚会和旅游，每次聚会都会照一张集体照上传到群共享，有一天，小A整理照片，想通过照片来看看他们这群基友的情况。我们假定，在同一张照片里同时出现的，两两之间都是好基友，基友的基友也是好基友。那么问题来了，你能帮小A确定任意的两个人是否好基友吗？
输入格式:
首先输入照片的张数N（N<10000），之后N行，按以下格式输入照片里的情况：

输入格式:
K P1 P2 P3 …… PK
其中K是照片中的人数，后面的K个数是照片中人的编号，我们假定从1开始连续编号，人数不超过10000。 接下来输出查询的个数T（T<10000），之后T行，每行输入两个数，分别是查询的两个人的编号。

输出格式:
首先输出基友圈的个数和所有人数，之后对T组查询，在一行中输出查询结果，如果两个是好基友，输出Yes，否则输出``No`。

输入样例:
在这里给出一组输入。例如：

4
3 11 1 2
3 3 4 10
4 1 5 7 8
3 9 6 12
2
10 5
7 11
输出样例:
在这里给出相应的输出。例如：

3 12
No
Yes

这个题就是一个普通并查集的问题，但是PTA最后一个测试点说我运行超时，爷也没有改出来，先堆在这里。

#include<iostream>
using namespace std;
#define M 20002
int n,m;
int pre[M],root[M],arr[1000],book[M];
void init()
{
	for(int i = 0; i < M; ++i)
	{
		pre[i] = i;
		root[i] = 0;
		book[i] = 0;
	}
}
int find(int x)
{
	if(x == pre[x]) 
		return x;
	else 
		return find(pre[x]);
}
void merge(int x,int y)
{
	int xx = find(x);
	int yy = find(y);
	if(xx != yy)
	{
		pre[yy] = xx;
	}
}
int main()
{
	int k;
	init();
	cin >> n;
	for(int i = 0; i < n; ++i)
	{
		cin >> k;
		for(int j = 0; j < k; ++j)
		{
			cin >> arr[j];
			book[arr[j]] = 1;
			if(j > 0) merge(arr[j - 1], arr[j]);
		}
	}
	int root_sum = 0,num_sum = 0;
	for(int i = 0; i < M; ++i)
	{
		if(book[i] == 1) 
		{
			root[find(i)] = 1;
		}
	}
	for(int i = 0; i < M; ++i)
	{
		root_sum += root[i];
		num_sum += book[i];
	}
	cin >> m;
	int a,b;
	cout << root_sum << " " << num_sum << endl;
	for(int i = 0; i < m; ++i)
	{
		cin >> a >> b;
		if(find(a) == find(b))  cout << "Yes" << endl;
		else	cout << "No" << endl;
	}
	return 0;
}