LA6910Cutting Tree 逆向并查集

先拆边，再倒着加边保存答案 注意记录某条边拆了几次就可以了

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
int f[20010],a[20010];
int ans[5010];
struct node
{
    char ch;
    int p,q;
};
node temp[20100];
int sum[20010];
int findset(int u)
{
    return f[u]==u?u:f[u]=findset(f[u]);
}
int n,k;
int main()
{
    int t;
    scanf("%d",&t);
    for(int ca=1;ca<=t;++ca){
        memset(sum,0,sizeof sum);
        printf("Case #%d:\n",ca);
        scanf("%d%d",&n,&k);
        for(int i = 0; i <= n; ++i) {
            f[i]=i;
        }
        for(int i = 1; i <= n; ++i) {
            scanf("%d",&a[i]);
            if(a[i]==0) a[i]=i;
            f[i]=a[i];
        }
        for(int i = k-1; i >= 0; --i) {
            scanf(" %c",&temp[i].ch);
            scanf(" %d",&temp[i].p);
            if(temp[i].ch=='Q') {
                scanf("%d",&temp[i].q);
            }
            else {
                f[temp[i].p]=temp[i].p;
                sum[temp[i].p]++;
            }
        }
        for(int i = 0; i < k; ++i) {
            if(temp[i].ch=='C') {
                ans[i]=-1;
                sum[temp[i].p]--;
                if(sum[temp[i].p]) continue;
                int x=findset(temp[i].p),y=findset(a[temp[i].p]);
                f[x]=y;
                if(y==0) f[x]=x;
            }
            else {
                int x=findset(temp[i].p),y=findset(temp[i].q);
                if(x==y) {
                    ans[i]=1;
                }
                else ans[i]=0;
            }
        }
        for(int i = k-1; i >= 0; --i) {
            if(ans[i]==0) {
                puts("NO");
            }
            else if(ans[i]==1) {
                puts("YES");
            }
        }
    }
    return 0;
}