并查集

需要注意的地方是并查集的路径压缩,最关键的东西是父节点数组,所有信息都存在这个上面。
这道题用了两个集合,第一个集合是用来存有多少个成员,第二个集合是用来存有多少个部落。其实也可用散列,但是集合的好处是可以直接输出集合的数量。

#include<bits/stdc++.h>
#include<iostream>
#include<set>
using namespace std;
const int maxn=1e4+5;
int arr[maxn];
int father[maxn];
int find(int x){
	if(x == father[x]) return x;
	else{
		int tmp = find(father[x]);
		father[x] = tmp;
		return tmp;
	}
}
void join(int x, int y){
	int fx = find(x), fy = find(y);
	if(fx != fy)
		father[fy] = fx;
} 
int main(){
	int n,k;
	set<int> st_total,st_group;
	cin>>n;
	for(int i=1;i<=maxn;i++){
		father[i]=i;
	}
	for(int i=1;i<=n;i++){
		cin>>k;
		for(int j=0;j<k;j++){
			cin>>arr[j];
			st_total.insert(arr[j]);
			if(j) join(arr[j-1], arr[j]); //j不为零的时候,合并两个元素 
		}
	}
	for(int i=1;i<=st_total.size();i++){
		st_group.insert(find(i));
	}
	cout<<st_total.size()<<" "<<st_group.size()<<endl;
	int m,u,v;
	cin>>m;
	for(int i=0;i<m;i++){
		cin>>u>>v;
		if(find(u)==find(v)) cout<<"Y"<<endl;
		else cout<<"N"<<endl;
	}
	
	return 0;
}