并查集 HEOI 2016/TJOI 2016 树

题意：一颗根为1的树，有两种操作
1.标记操作：对某个结点打上标记（在最开始，只有结点1有标记，其他结点均无标记，而且对于某个结点，可以打多次标记）

2.询问操作：询问某个结点最近的一个打了标记的祖先（这个结点本身也算自己的祖先）

询问祖先，我们会想到并查集，对于一个被打了标记的点，他的会管辖自己子树内的一些节点，如图：

图中1号，3号节点被标记，而红色的节点和1号节点是1号节点控制的，而绿色的节点和3号节点是3号节点控制的。

考虑并查集，如果我们按正常的顺序套并查集，那么我们只能将所有的点的$fat$都指向他最近的打了标记的祖先，无法对新标记的点及其控制的子树内的一些节点进行操作。

于是我们不妨反过来删标记，先将所有点打上标记，然后从后往前处理询问。如果是打标记，那么该点的标记-1，当一个点失去所有标记时，就将它的$fat$指向他父亲节点最近打了标记的祖先$（find(f[x])）$。如果是询问，直接将$find(x)$加入答案即可。因为是逆序，所以答案也逆序输出。

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int next,to;
}e[1001000];
int num,head[1001000],f[1001000],fat[1001000];
int book[1001000],g[1001000],ans[1001000];
char s[1001000];
void add(int from,int to)
{
    e[++num].next=head[from];
    e[num].to=to;
    head[from]=num;
}
int n,q;
int find(int x)
{
    if(x==fat[x])
        return x;
    else
    {
        fat[x]=find(fat[x]);
        return fat[x];
    }
}
void dfs(int x,int fa)
{
    if(!book[x])
        fat[x]=find(f[x]);
    for(int i=head[x];i;i=e[i].next)
    {
        int v=e[i].to;
        if(v==fa)
            continue;
        f[v]=x;
        dfs(v,x);
    }
}
int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n-1;++i)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    for(int i=1;i<=n;++i)
        fat[i]=i;
    book[1]=1;
    for(int i=1;i<=q;++i)
    {
        s[i]=getchar();
        while(s[i]!='Q'&&s[i]!='C')
            s[i]=getchar();
        scanf("%d",&g[i]);
        if(s[i]=='C')
            book[g[i]]++;
    }
    dfs(1,0);
    int k=0;
    for(int i=q;i>=1;--i)
    {
        if(s[i]=='C')
        {
            book[g[i]]--;
            if(!book[g[i]])
                fat[g[i]]=find(f[g[i]]);
        }
        else
            ans[++k]=find(g[i]);    
    }
    for(int i=k;i>=1;--i)
        printf("%d\n",ans[i]);
    return 0;
}