Description
给定一棵大小为 n 的有根点权树,支持以下操作: • 换根 • 修改点权 • 查询子树最小值
Input
第一行两个整数 n, Q ,分别表示树的大小和操作数。 接下来n行,每行两个整数f,v,第i+1行的两个数表示点i的父亲和点i的权。保证f < i。如 果f = 0,那么i为根。输入数据保证只有i = 1时,f = 0。 接下来 m 行,为以下格式中的一种: • V x y表示把点x的权改为y • E x 表示把有根树的根改为点 x • Q x 表示查询点 x 的子树最小值
Output
对于每个 Q ,输出子树最小值。
Sample Input
3 7
0 1
1 2
1 3
Q 1
V 1 6
Q 1
V 2 5
Q 1
V 3 4
Q 1
Sample Output
1
2
3
4
HINT
对于 100% 的数据:n, Q ≤ 10^5。
题解 以1为根建立dfs树，对于询问，考虑x与根的关系。
代码
#include<bits/stdc++.h> #define inf 1000000000 #define mod 1000000007 #define ll long long using namespace std; 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; } int n,Q,tim,tot,rt; int Head[100005],ret[200005],Next[200005]; int val[100005],l[100005],r[100005],q[100005],dep[100005]; int fa[100005][20],fac[20]; int mn[400005]; void ins(int u,int v){ret[++tot]=v;Next[tot]=Head[u];Head[u]=tot;} void dfs(int u) { tim++;l[u]=tim;q[tim]=u; for (int i=1;i<=16;i++) if (fac[i]<=dep[u]) fa[u][i]=fa[fa[u][i-1]][i-1]; else break; for (int i=Head[u];i;i=Next[i]) { fa[ret[i]][0]=u; dep[ret[i]]=dep[u]+1; dfs(ret[i]); } r[u]=tim; } void build(int k,int l,int r) { if (l==r){mn[k]=val[q[l]];return;} int mid=(l+r)>>1; build(k<<1,l,mid);build(k<<1|1,mid+1,r); mn[k]=min(mn[k<<1],mn[k<<1|1]); } void modify(int k,int l,int r,int x,int y) { if (l==r){mn[k]=y;return;} int mid=(l+r)>>1; if (x<=mid) modify(k<<1,l,mid,x,y); else modify(k<<1|1,mid+1,r,x,y); mn[k]=min(mn[k<<1],mn[k<<1|1]); } int query(int k,int l,int r,int x,int y) { if(x>y)return inf; if (l==x&&r==y) return mn[k]; int mid=(l+r)>>1; if (y<=mid) return query(k<<1,l,mid,x,y); else if (x>mid) return query(k<<1|1,mid+1,r,x,y); else return min(query(k<<1,l,mid,x,mid),query(k<<1|1,mid+1,r,mid+1,y)); } int main() { fac[0]=1;for (int i=1;i<20;i++) fac[i]=fac[i-1]<<1; n=read();Q=read(); for (int i=1;i<=n;i++) { int f=read();val[i]=read(); if (f) ins(f,i); } rt=1; dfs(1); build(1,1,n); char ch[5];int x; while (Q--) { scanf("%s",ch);x=read(); if (ch[0]=='V') { int val=read(); modify(1,1,n,l[x],val); } else if (ch[0]=='E') rt=x; else { if (rt==x) printf("%d\n",mn[1]); else if (l[x]<=l[rt]&&r[x]>=r[rt]) { int y=rt,d=dep[y]-dep[x]-1; for(int i=0;i<=16;i++) if(fac[i]&d)y=fa[y][i]; printf("%d\n",min(query(1,1,n,1,l[y]-1),query(1,1,n,r[y]+1,n))); } else printf("%d\n",query(1,1,n,l[x],r[x])); } } }