第十四届华中科技大学程序设计竞赛C—— Professional Manager

传送门：点击打开链接

题意：按照题目要求对树进行四种操作。

分析：初一看就觉得是个并查集的题，但是不知道怎么删点，后面看了别人的代码，才恍然大悟，加一个数组标记该点的编号就行了。另外还需要压缩路径，否则会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;
            }
        }
    }
}