BZOJ4551 [Tjoi2016&Heoi2016]树（洛谷P4092）

并查集

BZOJ题目传送门
洛谷题目传送门

先离线把所有标记都打上，然后倒着做，并查集维护当前节点的答案。如果是修改操作把当前节点并到它父亲上去，否则找一次爸爸就好了。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define F inline
using namespace std;
struct edge{ int nxt,to; }ed[N<<1];
int n,m,k,h[N],fa[N],ff[N],ans[N],q[N][2],f[N];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    return l==r?EOF:*l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)&&!isupper(ch)) ch=readc();
    if (isupper(ch)) return ch;
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
void dfs(int x,int FF){
    ff[x]=f[x]?x:FF;
    for (int i=h[x],v;i;i=ed[i].nxt)
        if ((v=ed[i].to)!=fa[x]) fa[v]=x,dfs(v,ff[x]);
}
#define add(x,y) ed[++k]=(edge){h[x],y},h[x]=k
int findfa(int x){ return x==ff[x]?x:ff[x]=findfa(ff[x]); }
int main(){
    n=_read(),m=_read();
    for (int i=1,x,y;i<n;i++)
        x=_read(),y=_read(),add(x,y),add(y,x);
    for (int i=1;i<=m;i++)
        q[i][0]=_read(),q[i][1]=_read(),f[q[i][1]]+=q[i][0]=='C';
    for (ff[1]++,fa[1]=1,dfs(1,1);m;m--){
        int x=q[m][1];
        if (q[m][0]=='Q') ans[++ans[0]]=findfa(x);
        else if (!(--f[x])) ff[x]=findfa(fa[x]);
    }
    for (;ans[0];ans[0]--) printf("%d\n",ans[ans[0]]);
    return 0;
}