1. 程式人生 > >bzoj 3306: 樹 dfs序+線段樹

bzoj 3306: 樹 dfs序+線段樹

題意

給定一棵大小為 n 的有根點權樹,支援以下操作:
換根
修改點權
查詢子樹最小值
n,q<=100000

分析

一開始還以為是某種神奇的lct姿勢,後來想了一下,發現如果當前根在x的子樹外的話x的子樹就還是原來的子樹,否則x當前的子樹就是除了x的根所在的子樹外的其他部分。
那麼直接上線段樹就好了。

程式碼

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int
N=100005; int n,m,dep[N],dfn[N],b[N],tim,a[N],fa[N][20],last[N],cnt,mn[N],mx[N]; struct edge{int to,next;}e[N*2]; struct tree{int mn;}t[N*4]; 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; } void addedge(int u,int v) { e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt; e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt; } void dfs(int x) { dep[x]=dep[fa[x][0]]+1;dfn[x]=++tim;mn[x]=tim;b[tim]=x; for (int i=1;i<=16;i++) fa[x][i]=fa[fa[x][i-1]][i-1
]; for (int i=last[x];i;i=e[i].next) { if (e[i].to==fa[x][0]) continue; fa[e[i].to][0]=x; dfs(e[i].to); } mx[x]=tim; } int get_lca(int x,int y) { if (dep[x]<dep[y]) swap(x,y); for (int i=16;i>=0;i--) if (dep[fa[x][i]]>=dep[y]) x=fa[x][i]; if (x==y) return x; for (int i=16;i>=0;i--) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i]; return fa[x][0]; } int get_bel(int x,int y) { for (int i=16;i>=0;i--) if (dep[fa[y][i]]>dep[x]) y=fa[y][i]; return y; } void updata(int d) { t[d].mn=min(t[d*2].mn,t[d*2+1].mn); } void build(int d,int l,int r) { if (l==r) { t[d].mn=a[b[l]]; return; } int mid=(l+r)/2; build(d*2,l,mid);build(d*2+1,mid+1,r); updata(d); } void ins(int d,int l,int r,int x,int y) { if (l==r) { t[d].mn=y; return; } int mid=(l+r)/2; if (x<=mid) ins(d*2,l,mid,x,y); else ins(d*2+1,mid+1,r,x,y); updata(d); } int query(int d,int l,int r,int x,int y) { if (x>y) return 1e9; if (l==x&&r==y) return t[d].mn; int mid=(l+r)/2; return min(query(d*2,l,mid,x,min(y,mid)),query(d*2+1,mid+1,r,max(x,mid+1),y)); } int main() { n=read();m=read(); for (int i=1;i<=n;i++) { int x=read();a[i]=read(); if (x) addedge(x,i); } int root=1; dfs(1); build(1,1,n); while (m--) { char op[2]; scanf("%s",op); if (op[0]=='V') { int x=read(),y=read(); ins(1,1,n,dfn[x],y); } else if (op[0]=='E') root=read(); else { int x=read(); if (x==root) printf("%d\n",query(1,1,n,1,n)); else if (get_lca(x,root)!=x) printf("%d\n",query(1,1,n,mn[x],mx[x])); else { int y=get_bel(x,root); printf("%d\n",min(query(1,1,n,1,mn[y]-1),query(1,1,n,mx[y]+1,n))); } } } return 0; }