bzoj4551 [Tjoi2016&Heoi2016]树（dfs序+线段树）

我们发现标记一个点只会影响他的子树的答案，用dfs序转化为区间覆盖即可，深度越深越接近。
还可以lct。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,dep[N],in[N],out[N],dfn=0;
vector<int>son[N];
struct node{
    int mx;
}tr[N<<2];
void dfs(int x){
    in[x]=++dfn;
    for(int i=0;i<son[x].size();++i){
        int y=son[x][i];dep[y]=dep[x]+1;dfs(y);
    }out[x]=dfn;
}
inline int Max(int x,int y){return dep[x]>dep[y]?x:y;}
inline void pushdown(int p){
    if(!tr[p].mx) return;
    tr[p<<1].mx=Max(tr[p<<1].mx,tr[p].mx);
    tr[p<<1|1].mx=Max(tr[p<<1|1].mx,tr[p].mx);tr[p].mx=0;
}
inline void change(int p,int l,int r,int x,int y,int val){
    if(x<=l&&r<=y){tr[p].mx=Max(tr[p].mx,val);return;}
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) change(p<<1,l,mid,x,y,val);
    if(y>mid) change(p<<1|1,mid+1,r,x,y,val);
}
inline int ask(int p,int l,int r,int x){
    if(l==r) return tr[p].mx?tr[p].mx:1;
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) return ask(p<<1,l,mid,x);
    return ask(p<<1|1,mid+1,r,x);
}
int main(){
//  freopen("tree.in","r",stdin);
//  freopen("tree.out","w",stdout);
    n=read();m=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();son[x].push_back(y);
    }dfs(1);
    while(m--){
        char op[5];scanf("%s",op+1);int x=read();
        if(op[1]=='C') change(1,1,n,in[x],out[x],x);
        else printf("%d\n",ask(1,1,n,in[x]));
    }return 0;
}