Cutting Tree（并查集）

题意：给你一个森林，q a b操作查询a和b之间能否到达，c d操作把d和他的爸爸的边删掉。

思路：并查集，所有点指向根节点，更新就行了。

//#pragma comment(linker, "/STACK:102400000,102400000")
#include <iostream>
#include<string.h>
#include<vector>
#include<queue>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<map>
#include<stdlib.h>
#include<time.h>
#include<stack>
#include<set>
#include<deque>
using namespace std;
typedef long long ll;
vector<int>tu[20010];
int f[20010],ff[20010];
void dfs(int now,int fa)
{
    f[now]=fa;
    for(int i=0,l=tu[now].size(); i<l; i++)
        dfs(tu[now][i],fa);
}
int main()
{
    int t,o=1;
    scanf("%d",&t);
    while(t--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        memset(f,0,sizeof(f));
        memset(ff,0,sizeof(ff));
        for(int i=1; i<=n; i++)
        {
            int tt;
            scanf("%d",&tt);
            tu[tt].push_back(i);
            if(!tt)
                ff[i]=f[i]=i;
            else
                ff[i]=f[i]=tt;
        }
        for(int i=1; i<=n; i++)
        {
            int tem=i;
            while(f[i]!=tem)
                tem=f[tem],f[i]=f[f[tem]];
            // cout<<"f["<<i<<"]="<<f[i]<<endl;
        }
        printf("Case #%d:\n",o++);
        while(k--)
        {
            char q[10];
            scanf("%s",q);
            if(q[0]=='C')
            {
                int d;
                scanf("%d",&d);
                for(int i=0,l=tu[ff[d]].size(); i<l; i++)
                    if(tu[ff[d]][i]==d)
                    {
                        tu[ff[d]].erase(tu[ff[d]].begin()+i);
                        break;
                    }
                dfs(d,d);
                //for(int i=1; i<=n; i++)
                //cout<<"f["<<i<<"]="<<f[i]<<endl;
            }
            else
            {
                int a,b;
                scanf("%d%d",&a,&b);
                if(f[a]==f[b])
                    puts("YES");
                else
                    puts("NO");
            }
        }
        for(int i=0; i<=n; i++)
            tu[i].clear();
    }
    return 0;
}