【数据结构 | C++】部落

在一个社区里，每个人都有自己的小圈子，还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里，于是要请你统计一下，在一个给定社区中，到底有多少个互不相交的部落？并且检查任意两个人是否属于同一个部落。

输入样例：
4
3 10 1 2
2 3 4
4 1 5 7 8
3 9 6 4
2
10 5
3 7
输出样例：
10 2
Y
N

#include<bits/stdc++.h>
using namespace std;
const int N=11000;
int a[N];
int find(int x)
{
	return a[x]==x?x:a[x]=find(a[x]);
}
void merge(int x,int y)
{
	int t1=find(x);
	int t2=find(y);
	if(t1!=t2)
	{
		a[t2]=t1;
	}
}
int main()
{
	int u=0;
	int cnt=0;
	queue<char > q; 
	set<int > s;
	int n,i,m;
	cin>>n;
	for(i=1;i<=10000;i++)
	{
		a[i]=i;
	}
	while(n--)
	{
		int t,x,y;
		cin>>t;
		if(t!=0)
		{
			cin>>x;
			if(u<x)
			{
				u=x;
			}
			s.insert(x);
			for(i=1;i<t;i++)
			{
				cin>>y;
				if(u<y)  
				{
					u=y;
				}
				merge(x,y); 
				s.insert(y); 
			}
		}
	}
	cin>>m;
	while(m--)
	{
		int x,y;
		cin>>x>>y;
		if(find(x)==find(y))
		{
			q.push('Y');
		}
		else
		{
			q.push('N');
		}
	}
	for(i=1;i<=u;i++)
	{
		if(a[i]==i)
		{
			cnt++;
		}
	}
	cout<<s.size()<<" "<<cnt<<endl;
	while(!q.empty())
	{
		cout<<q.front()<<endl;
		q.pop();
	}
	return 0;
}