Aizu - 2170 Marked Ancestor 并查集好题

传送门：AIZU2170

题意：给出一颗树，有两种操作：

1. mark  u  标记结点u

2.query  u  询问离u最近的且被标记的祖先结点是哪个

让你输出所有询问的和。

思路：刚读完题感觉是LCA一类的东西。。想到了离线处理，也想到了按标记和询问的先后搞事情，然后就没思路了。。

正解：将所有被标记的结点记录他们最早被标记的时间，然后将所有询问记录下来并倒着处理，因为如果我们倒着处理询问，那么我们处理完一个询问以后，时间上在这个询问之后的标记操作都可以忽略掉了，具体在树上怎么进行‘忽略’操作呢，这时候就要用到神奇的并查集了，并查集的路径压缩刚好可以做到这点，在处理一个询问的过程中，并查集寻根操作时会经过很多未标记点和很多‘标记过期’的点，这些点我们都可以直接忽略掉而让被询问点直接指向离他最近的并且标记尚未过期的点，这样就进行了树的重构，使得树高不断减小，从而降低了询问花费的时间。

感觉这个树重构的过程就是并查集路径压缩过程的完美体现。

代码：

#include<stdio.h>
#include<iostream>
#define ll long long
#define pb push_back
#define fi first
#define se second
#define pi acos(-1)
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,x,n) for(int i=x;i<n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
using namespace std;
typedef pair<int,int>P;
const int MAXN=100010;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int f[MAXN], mark[MAXN];
P Q[MAXN];
int getf(int k, int id)
{
	return mark[k] < id ? k : f[k] = getf(f[k], id);
}
int main()
{
	int n, q;
	while(~scanf("%d %d", &n, &q),n + q)
	{
		for(int i = 2; i <= n; i++)
		{
			scanf("%d", &f[i]);
			mark[i] = inf;
		}
		int cnt = 0;
		char c;
		int x;
		for(int i = 1; i <= q; i++)
		{
			scanf(" %c %d", &c, &x);
			if(c == 'Q')
			Q[cnt++] = P(i, x);
			else
			mark[x] = min(mark[x], i);
		}
		ll ans = 0;
		for(int i = cnt - 1; i >= 0; i--)
		ans += getf(Q[i].se, Q[i].fi);
		printf("%lld\n", ans);
	}
 	return 0;
}

越学越觉得并查集功能真的很强大。