传送门:点击打开链接
题意:按照题目要求对树进行四种操作。
分析:初一看就觉得是个并查集的题,但是不知道怎么删点,后面看了别人的代码,才恍然大悟,加一个数组标记该点的编号就行了。另外还需要压缩路径,否则会tle。所谓压缩路径就是把这一团的点直接指向祖先,大大减少find函数查找的时间。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
int p[N],id[N],ct[N],n,q,t;
void init(){
for(int i=1;i<N;i++){
p[i] = id[i] = i;
ct[i]=1;
}
}
int f(int x){
return x==p[x] ? x : p[x]=f(p[x]);
}
int main(){
ios::sync_with_stdio(false);
cin>>t;
int cas=1;
while(t--){
cout<<"Case #"<<cas++<<":"<<endl;
cin>>n>>q;
init();
while(q--){
int op,u,v;
cin>>op;
if(op==1){
cin>>u>>v;
int x=f(id[u]),y=f(id[v]);
if(x!=y){
p[x]=y;
ct[y]+=ct[x];
}
}else if(op==2) {
cin>>u;
ct[f(id[u])]--;
id[u]=++n;
}else if(op==3) {
cin>>u;
cout<<ct[f(id[u])]<<endl;
}else {
cin>>u>>v;
if(f(id[u])==f(id[v])) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
}
}
}