「BZOJ3306」树
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。
题解
摘自zyf的blog
我们先以1为根dfs并建立倍增数组,然后如果根换成了rt,然后要查询x子树内的最小值。我们分情况讨论:
1)若x==rt,则直接输出整棵树的最小值
2)若lca(x,rt)既不等于x那么直接输出x的子树内的最小值
3)若lca(x,rt)==x那么我们发现整棵树除了x向下走可以到达rt的子树之外全部成了x在rt为根下的子树,那我们把这棵子树中最接近x的节点y求出,在整个区间中踢掉y在1根下子树的范围即可。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
#include<iostream> #include<cstring> #include<cstdio> #include<algorithm> #include<cmath> #define ll long long #define inf 1000000000 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 bin[20]; int n,Q,ind,root,cnt; int val[100005],l[100005],r[100005],q[100005],deep[100005]; int fa[100005][20]; struct edge{ int to,next; }e[100005];int last[100005]; struct seg{ int l,r,mn; }t[400005]; void insert(int u,int v) { e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt; } void update(int k) { t[k].mn=min(t[k<<1].mn,t[k<<1|1].mn); } void build(int k,int l,int r) { t[k].l=l;t[k].r=r; if(l==r){t[k].mn=val[q[l]];return;} int mid=(l+r)>>1; build(k<<1,l,mid);build(k<<1|1,mid+1,r); update(k); } void modify(int k,int pos,int val) { int l=t[k].l,r=t[k].r,mid=(l+r)>>1; if(l==r) { t[k].mn=val;return; } if(pos<=mid)modify(k<<1,pos,val); else modify(k<<1|1,pos,val); update(k); } int query(int k,int x,int y) { if(x>y)return inf; int l=t[k].l,r=t[k].r,mid=(l+r)>>1; if(l==x&&y==r)return t[k].mn; if(y<=mid)return query(k<<1,x,y); else if(x>mid)return query(k<<1|1,x,y); return min(query(k<<1,x,mid),query(k<<1|1,mid+1,y)); } void dfs(int x) { l[x]=++ind;q[ind]=x; for(int i=1;i<=16;i++) if(bin[i]<=deep[x])fa[x][i]=fa[fa[x][i-1]][i-1]; else break; for(int i=last[x];i;i=e[i].next) { fa[e[i].to][0]=x; deep[e[i].to]=deep[x]+1; dfs(e[i].to); } r[x]=ind; } int main() { bin[0]=1;for(int i=1;i<20;i++)bin[i]=bin[i-1]<<1; n=read(),Q=read(); for(int i=1;i<=n;i++) { int f=read();val[i]=read(); if(f)insert(f,i); } dfs(root=1); build(1,1,n); char ch[5];int x; for(int i=1;i<=Q;i++) { scanf("%s",ch);x=read(); if(ch[0]=='V') { int val=read(); modify(1,l[x],val); } else if(ch[0]=='E')root=x; else { if(root==x)printf("%d\n",t[1].mn); else if(l[x]<=l[root]&&r[x]>=r[root]) { int y=root,d=deep[y]-deep[x]-1; for(int i=0;i<=16;i++) if(bin[i]&d)y=fa[y][i]; printf("%d\n",min(query(1,1,l[y]-1),query(1,r[y]+1,n))); } else printf("%d\n",query(1,l[x],r[x])); } } return 0; } |
Subscribe