COGS 2274. [HEOI 2016] tree

本来想写一道搜索，结果让题目列表给骗了、、、

树剖可做

但是10min的并查集更好

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 200010
using namespace std;
int n,head[maxn],cnt;
struct opt{int tp,x;}q[maxn];
struct Edge{ int to,nxt; }e[maxn<<1];
void add(int u, int v){
    cnt ++;e[cnt].to=v;
    e[cnt].nxt=head[u];head[u]=cnt;
}
int mark[maxn],pre[maxn],fa[maxn],ans[maxn];
int find(int x){ return x == pre[x] ? x : pre[x] = find(pre[x]); }
void dfs(int u){
    if(mark[u]) pre[u]=u;
    else pre[u]=find(fa[u]);
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa[u])continue;
        fa[v]=u;
        dfs(v);
    }
}
int main(){
    freopen("heoi2016_tree.in","r",stdin);
    freopen("heoi2016_tree.out","w",stdout);
    int m,u,v;
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    mark[1]++;
    char cmd[5];
    for(int i=1;i<=m;i++){
        scanf("%s%d",cmd,&q[i].x);
        if(cmd[0] == 'C'){
            q[i].tp=1;//查询还是修改标记 
            mark[q[i].x]++;
        }
        else q[i].tp=0;
    }
    dfs(1);
    int x;
    for(int i=m;i>=1;i--){
        x=q[i].x;
        if(q[i].tp){
            mark[x]--;
            if(mark[x]==0) pre[x]=find(fa[x]);
        }
        else ans[i]=find(x);
    }
    for(int i=1;i<=m;i++)
        if(q[i].tp==0) printf("%d\n", ans[i]);
    return 0;
}