【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 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]));
        } 
    }
}