题意:给一棵树,节点是1到n的编号,1是根,最开始被标记,进行若干次询问,每次询问可将某个节点标记,也可以给出节点编号去找离它最近的被标记的编号;输出最后所有寻找的询问中的编号和。
利用并查集思想,刚开始节点1设为父亲,之后被标记就把当前节点设为父亲,找最近编号就直接find它的父亲,加到一起输出。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<set>
#include<bitset>
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<utility>
#define INF 1000000007
#define inf 100000000000000000
using namespace std;
typedef long long ll;
int far[100005],n,q,i,j,x;
ll sum;
char c;
int findx(int k)
{
if(far[k]==k)return k;
findx(far[k]);
}
int main()
{
while(scanf("%d%d",&n,&q)!=EOF&&(n!=0||q!=0))
{
sum=0;
for(i=2;i<=n;i++)scanf("%d",&far[i]);
far[1]=1;
for(i=0;i<q;i++)
{
scanf("\n%c%d",&c,&x);
if(c=='Q')sum+=findx(x);
else far[x]=x;
}
printf("%lld\n",sum);
}
return 0;
}