并查集入门,关键是想到并查集,我还想了其他操作好久发现没有什么想法
一个合并一个查找,就该想到并查集嘛
对于每一棵树上的鸟,可以将第一只作为已知在树上的鸟,把后面的鸟捉过来即可
#include <bits/stdc++.h>
using namespace std;
int fa[102000];
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);
}
void Union(int a,int b){int fx=find(a);int fy=find(b);if(fx!=fy) fa[fy]=fx;
}
set<int>s;
int main(){int n;cin>>n;for(int i=1;i<=101000;i++) fa[i]=i;for(int i=1;i<=n;i++){int k;cin>>k;int boss;cin>>boss;s.insert(boss);for(int j=1;j<k;j++){int another_bird;cin>>another_bird;s.insert(another_bird); Union(boss,another_bird);}}int ans=0;for(int i=1;i<=s.size();i++){if(find(i)==i) ans++;}printf("%d %d\n",ans,s.size());int q;cin>>q;while(q--){int x,y;cin>>x>>y;if(find(x)!=find(y)){printf("No\n");}else{printf("Yes\n");}}return 0;
}