【bzoj3306】【樹】【dfs序+線段樹】
阿新 • • 發佈:2019-02-12
對於 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); } } } }