bzoj4551[Tjoi2016&Heoi2016]树

bzoj4551[Tjoi2016&Heoi2016]树

题意:

给个根节点为1的n点树，初始时节点1标记，Q个操作，每次可以标记一个点或求一个点最近一个标记了的祖先。

题解：

链剖可以写，当正解应该是并查集。离线读入所有操作，累加每个节点的标记次数，之后所有未被标记的节点向其父亲节点连边，然后倒着来，如果操作是询问则输出这个节点在并查集中的根节点，如果是标记则将该节点的标记数减1，一旦这个节点的标记数减到了0，就让它向父亲节点连边。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#define inc(i,j,k) for(int i=j;i<=k;i++)
#define dec(i,j,k) for(int i=j;i>=k;i--)
#define maxn 100100
using namespace std;

inline int read(){
    char ch=getchar(); int f=1,x=0;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();} while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return f*x;
}
struct e{int t,n;}; e es[maxn*2]; int ess,g[maxn];
void pe(int f,int t){
    es[++ess]=(e){t,g[f]}; g[f]=ess; es[++ess]=(e){f,g[t]}; g[t]=ess;
}
struct ask{char opt[3];int x;}; ask asks[maxn];
int n,fa[maxn],p[maxn],q,tag[maxn];
void dfs(int x,int f){for(int i=g[x];i;i=es[i].n)if(es[i].t!=f)fa[es[i].t]=x,dfs(es[i].t,x);}
int find(int x){return x==p[x]?x:p[x]=find(p[x]);}
void uni(int x){int y=find(fa[x]),z=find(x); if(y!=z)p[z]=y;}
int main(){
    n=read(); q=read();
    inc(i,1,n-1){int a=read(),b=read(); pe(a,b);} dfs(1,0); tag[1]++;
    inc(i,1,q){scanf("%s",asks[i].opt); asks[i].x=read(); if(asks[i].opt[0]=='C')tag[asks[i].x]++;}
    inc(i,1,n)p[i]=i; inc(i,1,n)if(!tag[i])uni(i);
    dec(i,q,1){
        if(asks[i].opt[0]=='C'){
            tag[asks[i].x]--;
            if(!tag[asks[i].x])uni(asks[i].x);
        }
        if(asks[i].opt[0]=='Q')asks[i].x=find(asks[i].x);
    }
    inc(i,1,q)if(asks[i].opt[0]=='Q')printf("%d\n",asks[i].x);
    return 0;
}

 

20160611