bzoj3306（线段树,dfs序）

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。
　　

---

查询子树信息结合到dfs序
唯一麻烦的是换根的操作
考虑当前根$root$与$x$的关系
如果$x=root$，直接查询整个树
如果$lca(x,root) = x，x$的子树相当于是整棵树切掉x的root存在的这棵子树,查询dfs序上整个区间去掉这个区间即可
如果$lca(x,root)!=x$ 直接查询的子树即可

#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define Pbi pair<bool,int>
#define pb push_back 
#define ll long long
#define ls root * 2
#define rs root * 2 + 1
const int INF = 1e9;
int n , q , linkk[101000] , a[101000];
int dfs[101000] , mn[401000] , fa[101000][20] , in[101000] , out[101000];
int root , t , tot;
int M_dep , dep[101000];
struct node{
    int n , y;
}e[201000];
int read()
{
    int sum = 0;char c = getchar();bool flag = true;
    while( c < '0' || c > '9' ) {if(c == '-') flag = false;c = getchar();}
    while( c >= '0' && c <= '9' ) sum = sum * 10 + c - 48 , c = getchar();
    if(flag)  return sum;
     else return -sum;
}  
void insert(int x,int y)
{
    e[++t].y = y;e[t].n = linkk[x];linkk[x] = t;
    e[++t].y = x;e[t].n = linkk[y];linkk[y] = t;
}
void build(int root,int l,int r)
{
    if(l == r)
    {
        mn[root] = a[dfs[l]];
        return;
    }
    int mid = (l + r)>>1;
    build(ls,l,mid);build(rs,mid+1,r);
    mn[root] = min(mn[ls] , mn[rs]);
}
void change(int root,int l,int r,int x,int y)
{
    if(l == r)
    {
        mn[root] = y;
        return;
    }
    int mid = (l + r)>>1;
    x<=mid ? change(ls,l,mid,x,y) : change(rs,mid+1,r,x,y);
    mn[root] = min(mn[ls] , mn[rs]);
}
int query(int root,int l,int r,int x,int y)
{
    if(x > y) return INF;
    if(r < x || l > y) return INF;
    if(x <= l && r <= y) return mn[root];
    int mid = (l + r)>>1;
    return min( query(ls,l,mid,x,y) , query(rs,mid+1,r,x,y) );
}
void get(int x,int deep)
{
    dep[x] = deep;dfs[++tot] = x;in[x] = tot;
    rept(i,x)
        if(e[i].y != fa[x][0])
            get(e[i].y,deep+1);
    out[x] = tot;
    M_dep = max(M_dep,deep);
}
void init()
{
    n = read();q = read();
    rep(i,1,n)
    {
        fa[i][0] = read() , a[i] = read();
        if(fa[i][0] != 0) insert(i,fa[i][0]);
    }
    get(1,1);
    for(int i = 1;(1<<i) <= M_dep;++i)
        rep(j,1,n)
            fa[j][i] = fa[fa[j][i-1]][i-1];
    return;
}
Pbi check(int x)//判断x是不是root的祖先
{
    if(dep[x] > dep[root]) return make_pair(false,0);
    int k = log2(dep[root] - dep[x]) + 1;
    int y = root;
    repp(i,k,0)
        if( (dep[y] - (1<<i)) > dep[x] )
            y = fa[y][i];
    if(fa[y][0] == x) return make_pair(true,y);
    else return make_pair(false,0);
}
int main()
{
    init(); build(1,1,n);
    root = 1;
    rep(i,1,q)
    {
        char c = getchar();
        while(c != 'V' && c != 'E' && c != 'Q') c = getchar();
        if(c == 'E') 
        {
            int x = read();
            root = x;
        }
        else 
        if(c == 'V')
        {
            int x = read() , y = read();
            a[x] = y;
            change(1,1,n,in[x],y);
        }
        else
        if(c == 'Q')
        {
            int x = read();
            if(x == root)
            {
                printf("%d\n",query(1,1,n,1,n));
            }
            else
            {
                Pbi ans = check(x);
                int y = ans.second;
                if(ans.first) printf("%d\n",min(query(1,1,n,1,in[y]-1) , query(1,1,n,out[y]+1,n) ) );
                else printf("%d\n",query(1,1,n,in[x],out[x]));
            }

        }
    }
    return 0;

}