并查集 AOJ 2170

题意:
有一个树,有些节点染色,每次有两种操作,第一,统计该节点到离它最近的染色父亲结点的的数值,第二,为某一个节点染色
题解:
直接记录了每一个节点的父亲节点,每一次搜索就计算到他的染色父亲结点为止

#include 
#include
#define MAX_N 100005
using namespace std;
typedef long long ll;
typedef pair<int ,bool> p;
p fa[MAX_N];
int main(){
    int n,q;
    fa[1].first = 1;
    fa[1].second = true;
    while(~scanf("%d%d",&n,&q)&&n>0){
        ll ans = 0;
        for(int i = 2;i <= n;i++){
            scanf("%d",&fa[i].first);
            fa[i].second = false;
        }
        getchar();
        for(int i = 0;ichar ch;int a;
            scanf("%c%d",&ch,&a);
            getchar();
            if(ch =='Q'){
                while(fa[a].second==false){
                    a = fa[a].first;
                }
                ans += a;
            }
            else{
                fa[a].second = true;
            }

        }
        printf("%lld\n",ans);
    }
}

你可能感兴趣的:(集训-挑战-并查集)