BZOJ4551 HEOI2016树

可以用并查集，记忆化搜索，线段树多种方法实现。

我这里写的是依照dfs序建线段树，维护区间最大值。

#include<bits/stdc++.h>
using namespace std;
const int N=100005;
int head[N],lz[N<<2],t[N<<2],id[N],end[N],cnt,idx,pos[N];
struct node
{
    int to,nex;
}e[N<<1];
void add(int x,int y)
{
    e[++cnt].to=y;e[cnt].nex=head[x];head[x]=cnt;
}
void dfs(int x,int fa)
{
    id[x]=++idx;pos[idx]=x;
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(y==fa)continue;
        dfs(y,x);
    }
    end[x]=idx;
}
void pushdown(int x)
{
    if(lz[x])
    {
        t[x<<1]=max(t[x<<1],lz[x]);
        t[x<<1|1]=max(t[x<<1|1],lz[x]);
        lz[x<<1]=max(lz[x],lz[x<<1]);
        lz[x<<1|1]=max(lz[x],lz[x<<1|1]);
        lz[x]=0;
    }
}
void update(int p,int l,int r,int L,int R,int w)
{
    if(l==L&&r==R)
    {
        t[p]=max(t[p],w);lz[p]=max(lz[p],w);return;
    }
    int mid=l+r>>1;pushdown(p);
    if(R<=mid)update(p<<1,l,mid,L,R,w);
    else if(L>mid)update(p<<1|1,mid+1,r,L,R,w);
    else update(p<<1,l,mid,L,mid,w),update(p<<1|1,mid+1,r,mid+1,R,w);
}
int query(int p,int l,int r,int pos)
{
    if(l==r)return t[p];
    int mid=l+r>>1;pushdown(p);
    if(pos<=mid)return query(p<<1,l,mid,pos);
    else if(pos>mid)return query(p<<1|1,mid+1,r,pos);
}
int main()
{
    //freopen("tree.in","r",stdin);
    //freopen("tree.out","w",stdout);
    int n,q;scanf("%d%d",&n,&q);
    for(int i=1;i<n;++i)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    update(1,1,n,id[1],end[1],1);
    for(int i=1;i<=q;++i)
    {
        char s[3];int x;scanf("%s",s);scanf("%d",&x);
        if(s[0]=='C')
        {
            update(1,1,n,id[x],end[x],id[x]);
        }
        else
        {
            printf("%d
",pos[query(1,1,n,id[x])]);
        }
    }
        return 0;
}