传送门:点击打开链接
题意:按照题目要求对树进行四种操作。
分析:初一看就觉得是个并查集的题,但是不知道怎么删点,后面看了别人的代码,才恍然大悟,加一个数组标记该点的编号就行了。另外还需要压缩路径,否则会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; } } } }