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