BZOJ4551 HEOI2016树

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

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

 #include<bits/stdc++.h>
 using namespace std;
 const int N=;
 int head[N],lz[N<<],t[N<<],id[N],end[N],cnt,idx,pos[N];
 struct node
 {
     int to,nex;
 }e[N<<];
 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<<]=max(t[x<<],lz[x]);
         t[x<<|]=max(t[x<<|],lz[x]);
         lz[x<<]=max(lz[x],lz[x<<]);
         lz[x<<|]=max(lz[x],lz[x<<|]);
         lz[x]=;
     }
 }
 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>>;pushdown(p);
     if(R<=mid)update(p<<,l,mid,L,R,w);
     else if(L>mid)update(p<<|,mid+,r,L,R,w);
     else update(p<<,l,mid,L,mid,w),update(p<<|,mid+,r,mid+,R,w);
 }
 int query(int p,int l,int r,int pos)
 {
     if(l==r)return t[p];
     int mid=l+r>>;pushdown(p);
     if(pos<=mid)return query(p<<,l,mid,pos);
     else if(pos>mid)return query(p<<|,mid+,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=;i<n;++i)
     {
         int x,y;
         scanf("%d%d",&x,&y);
         add(x,y);
         add(y,x);
     }
     dfs(,);
     update(,,n,id[],end[],);
     for(int i=;i<=q;++i)
     {
         char s[];int x;scanf("%s",s);scanf("%d",&x);
         if(s[]=='C')
         {
             update(,,n,id[x],end[x],id[x]);
         }
         else
         {
             printf("%d\n",pos[query(,,n,id[x])]);
         }
     }
         return ;
 }