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;
}