先拆边,再倒着加边保存答案 注意记录某条边拆了几次就可以了
#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;
}