Aizu2170 Marked Ancestor（并查集）

https://vjudge.net/problem/Aizu-2170

并查集用于管理元素分组情况。

建树pre[]记录父节点，一开始只有结点1被标记了，所以find()最终得到的根都是1.

如果遇到M操作，即将树断开（很神奇的操作）。

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stack>
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
#define IO ios::sync_with_stdio(false);cin.tie(0);
#define INF 0x3f3f3f3f
typedef unsigned long long ll;
using namespace std;
int n, q, m, pre[100010];
char c;
int find(int x)
{
    while(x != pre[x]){
        x = pre[x];
    }
    return x;
}
int main()
{
    IO;
    while(cin >> n >> q){ 
        if(!n&&!q) break;
        ll sum=0;
        pre[1] = 1;
        for(int i = 2; i <= n; i++){
            cin >> pre[i];
        }
        for(int i = 0; i < q; i++){
            cin >> c >> m;
            if(c == 'Q') 
                sum += find(m);
            else {
                pre[m] = m;//断开 
            }
        }
        cout << sum << endl;
    }
    return 0;
}