【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。

题解
以1为根建立dfs树，对于询问，考虑x与根的关系。

代码

#include<bits/stdc++.h>#define inf 1000000000#define mod 1000000007#define ll long longusing 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]));        }     }}