树 bzoj4551洛谷4092 tjoi2016&heoi2016

题意：给一棵根是1的树，有两种操作，第一种是给一个节点打上标记，第二种是询问一个点最近的有标记的祖先。

这个题据说可以dfs序+线段树，但是我并不会，所以我用的是一种并查集的方法。

本题可以类比星球大战，都是逆向使用并查集，所以本题也是离线操作，因为同一个点会被重复打标记，所以要记录一下每个点最终被标记了多少次。如果一个节点被标记了，那么把它的父节点改为它自己。倒着往回做时每一有一个点的标记数变为0，就将它和它的原来的父节点相连即可

代码：

update：之前的代码不知道为什么在洛谷上最后一个点有时候会输出0，我没有想好原因，于是写了个特判。如果有知道原因的希望能赐教，谢谢。（现在代码在洛谷已经改成可以过的了）

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
 
int n,q,fa[200010],tag[200020],ask[200010],cask,ctag,ji[200010];
int book[200010],cnt,f[200010],j[200010];
int getr(int x)
{
    if(x==f[x])
    return x;
    else
    {
        f[x]=getr(f[x]);
        return f[x];
    }
}
int main()
{
    scanf("%d%d",&n,&q);
    fa[1]=1;
    f[1]=1;
    j[1]=1;
    for(int i=1;i<=n-1;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        fa[v]=u;
        f[v]=u;
    }
    for(int i=1;i<=q;i++)
    {
        char s;
        int x;
        cin>>s>>x;
        if(s=='Q')
        {
            ask[++cask]=i;
            book[i]=x;
        }
        if(s=='C')
        {
            tag[++ctag]=i;
            f[x]=x;
            book[i]=x;
            j[x]++;
        }
    }
    for(int i=q;i>=1;i--)
    {
        if(tag[ctag]==i)
        {
            j[book[i]]--;
            if(j[book[i]]==0)
            f[book[i]]=fa[book[i]];
            ctag--;
        }
        if(ask[cask]==i)
        {
            ji[++cnt]=getr(book[i]);
            cask--;
        }
    }
    for(int i=cnt;i>=1;i--)
    {
    	if(ji[i]==0)
    	printf("1\n");
    	else
        printf("%d\n",ji[i]);
    }
    return 0;
}