1. 程式人生 > >【bzoj3306】【樹】【dfs序+線段樹】

【bzoj3306】【樹】【dfs序+線段樹】

  對於 100% 的資料:n, Q ≤ 10^5。

題解:首先按dfs序建線段樹。

           對於換根操作。我們分三種情況討論(設root為當前根,x為要查詢的子樹)

           1.若root==x直接查詢整棵樹即可。

           2.若lca(root,x)!=x那就直接查詢x這個子樹即可。

           3.若lca(root,x)=x那就查詢和x相連的和root最近的點的子樹的補集即可。

畫個圖看看就明白了。

程式碼:

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 100010
using namespace std;
struct use{int st,en;}e[N*2];
int point[N],next[N*2],now,fa[N][30],v[N],tree[N*4],mx[N],n,q;
int d[N],pos[N],p,cnt,deep[N],x,y,f;
char ch[10];
inline int read()
{
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
void add(int x,int y){
  next[++cnt]=point[x];point[x]=cnt;
  e[cnt].st=x;e[cnt].en=y; 
}
void dfs(int x){
  for (int i=1;(1<<i)<=deep[x];i++) fa[x][i]=fa[fa[x][i-1]][i-1];
  d[x]=mx[x]=++p;
  for (int i=point[x];i;i=next[i])
    if(e[i].en!=fa[x][0]){
      fa[e[i].en][0]=x;deep[e[i].en]=deep[x]+1;
      dfs(e[i].en);mx[x]=max(mx[e[i].en],mx[x]);    
    } 
}
int lca(int x,int y){
  if (deep[x]<deep[y]) swap(x,y);int t=deep[x]-deep[y];
  for (int i=0;i<=20;i++) if (t&(1<<i)) x=fa[x][i];
  for (int i=20;i>=0;i--) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
  if (x==y) return x;else return fa[x][0];
}
void change(int k,int l,int r,int p,int v){
   int mid=(l+r)>>1;
   if (l==r){tree[k]=v;return;}
   if (p<=mid) change(k<<1,l,mid,p,v);
   else change(k<<1|1,mid+1,r,p,v);
   tree[k]=min(tree[k<<1],tree[k<<1|1]);
}
int query(int k,int l,int r,int ll,int rr){
  int mid=(l+r)>>1,ans(999999999);
  if (ll<=l&&r<=rr) return tree[k];
  if (ll<=mid) ans=min(ans,query(k<<1,l,mid,ll,rr));
  if (rr>mid) ans=min(ans,query(k<<1|1,mid+1,r,ll,rr));
  return ans;
} 
int cal(int x,int t){
  for (int i=0;i<=20;i++)if ((1<<i)&t) x=fa[x][i];
  return x;
}
int main(){
  n=read();q=read();now=1;
  for (int i=1;i<=n;i++){
   f=read();v[i]=read();
   if (f){add(f,i);add(i,f);}   
  }
  dfs(1);memset(tree,127/3,sizeof(tree));   
  for (int i=1;i<=n;i++) change(1,1,n,d[i],v[i]);
  for (int i=1;i<=q;i++){
    scanf("%s",&ch); 
    if (ch[0]=='V'){x=read();y=read();change(1,1,n,d[x],y);}
    if (ch[0]=='E'){x=read();now=x;}
    if (ch[0]=='Q'){
      x=read();f=lca(now,x);
      if (now==x) {printf("%d\n",query(1,1,n,1,mx[1]));continue;}
      if (f!=x) printf("%d\n",query(1,1,n,d[x],mx[x]));
      if (f==x){
        int u=cal(now,deep[now]-deep[x]-1);
        int ans=query(1,1,n,1,d[u]-1);
        if (mx[u]<n) ans=min(ans,query(1,1,n,mx[u]+1,n));
        printf("%d\n",ans);
      }
    }
  }
}