bzoj3083 遥远的国度 树链剖分+线段树

题目大意：
维护一棵树，支持三个操作：
1、换根；
2、链修改；
3、查询子树最小值。

题目分析：
如果只有2、3操作就可以直接树链剖分+线段树裸上了。
加上1操作后，我们仍然可以树链剖分+线段树裸上。

对于查询操作，我们分三种情况讨论：
假设查询的点为x。
1、根和x相同：直接输出整棵树的最小值。
2、根不在x的子树中：直接查询原x的子树。
3、根在x的子树中：先找到根在x的哪个子树里，把这个子树刨掉，然后查寻剩余所有点的最小值，那么我们就找到x到根路径上的第一个点，把这个点为根的子树刨掉就可以了。这个点可以倍增求。

代码如下：

#include <cstdio>
#include <algorithm>
#include <iostream>
#define ls(c) (c<<1)
#define rs(c) (c<<1|1)
#define N 120000
using namespace std;
inline int Min(int x,int y) { return x<y?x:y; }
int n,m,Capital;
int fir[N],nes[N<<1],v[N<<1],tot=1;
int q[N],fa[N],dep[N],sz[N],son[N],ld[N],xl[N],pos[N],top;
int st[N][18];
struct segment{
    int l,r,val,mark;
}seg[N<<2];
void edge(int x,int y)
{
    v[++tot]=y;
    nes[tot]=fir[x];
    fir[x]=tot;
}
#define edge(x,y) edge(x,y),edge(y,x)
void dfs1(int c)
{
    sz[c]=1; son[c]=0;
    dep[c]=dep[fa[c]]+1;
    st[c][0]=fa[c];
    for(int i=1;i<=17;i++)
        st[c][i]=st[st[c][i-1]][i-1];
    for(int t=fir[c];t;t=nes[t])
    {
        if(fa[c]==v[t]) continue;
        fa[v[t]]=c;
        dfs1(v[t]);
        sz[c]+=sz[v[t]];
        if(sz[v[t]]>=sz[son[c]]) son[c]=v[t];
    }
}
void dfs2(int c)
{
    xl[++top]=c;
    pos[c]=top;
    ld[c]=c;
    if(son[fa[c]]==c) ld[c]=ld[fa[c]];
    if(son[c]) dfs2(son[c]);
    for(int t=fir[c];t;t=nes[t])
    {
        if(fa[c]==v[t] || son[c]==v[t]) continue;
        dfs2(v[t]);
    }
}
void build_tree(int c,int l,int r)
{
    seg[c].l=l; seg[c].r=r;
    seg[c].mark=-1;
    if(l==r)
    {
        seg[c].val=q[xl[l]];
        return;
    }
    int mid=l+r>>1;
    build_tree(ls(c),l,mid);
    build_tree(rs(c),mid+1,r);
    seg[c].val=Min(seg[ls(c)].val,seg[rs(c)].val);
}
void add_mark(int c,int v)
{
    seg[c].val=v;
    seg[c].mark=v;
}
void push_down(int c)
{
    if(~seg[c].mark && seg[c].l!=seg[c].r)
    {
        add_mark(ls(c),seg[c].mark);
        add_mark(rs(c),seg[c].mark);
    }
    seg[c].mark=-1;
}
void update(int c,int l,int r,int v)
{
    push_down(c);
    if(l<=seg[c].l && r>=seg[c].r)
    {
        add_mark(c,v);
        return;
    }
    int mid=seg[c].l+seg[c].r>>1;
    if(l<=mid) update(ls(c),l,r,v);
    if(r>mid)  update(rs(c),l,r,v);
    seg[c].val=Min(seg[ls(c)].val,seg[rs(c)].val);
}
int query(int c,int l,int r)
{
    push_down(c);
    if(l<=seg[c].l && r>=seg[c].r) return seg[c].val;
    int mid=seg[c].l+seg[c].r>>1;
    if(r<=mid) return query(ls(c),l,r);
    if(l>mid) return query(rs(c),l,r);
    return Min(query(ls(c),l,r),query(rs(c),l,r));
}
void change(int x,int y,int z)
{
    while(ld[x]!=ld[y])
    {
        if(dep[ld[x]]<dep[ld[y]]) swap(x,y);
        update(1,pos[ld[x]],pos[x],z);
        x=fa[ld[x]];
    }
    if(dep[x]<dep[y]) swap(x,y);
    update(1,pos[y],pos[x],z);
}
int Lca(int x,int y)
{
    while(ld[x]!=ld[y])
    {
        if(dep[ld[x]]<dep[ld[y]]) swap(x,y);
        x=fa[ld[x]];
    }
    if(dep[x]<dep[y]) swap(x,y);
    return y;
}
int exLca(int x,int LCA)
{
    for(int i=17;i>=0;i--)
        if(dep[st[x][i]]>dep[LCA])
            x=st[x][i];
    return x;
}
int query(int l,int r)
{
    if(l>r) return 2147483647;
    return query(1,l,r);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        edge(x,y);
    }
    for(int i=1;i<=n;i++) scanf("%d",&q[i]);
    scanf("%d",&Capital);
    dfs1(1); dfs2(1);
    build_tree(1,1,n);
    for(int i=1,x,y,opt,z,LCA;i<=m;i++)
    {
        scanf("%d",&opt);
        switch(opt)
        {
        case 1:
            scanf("%d",&x);
            Capital=x;
            break;
        case 2:
            scanf("%d%d%d",&x,&y,&z);
            change(x,y,z);
            break;
        case 3:
            scanf("%d",&x);
            if(Capital==x) { printf("%d\n",seg[1].val); break; }
            LCA=Lca(Capital,x);
            if(LCA==Capital || LCA!=x) printf("%d\n",query(pos[x],pos[x]+sz[x]-1));
            else
            {
                x=exLca(Capital,LCA);
                printf("%d\n",Min(query(1,pos[x]-1),query(pos[x]+sz[x],n)));
            }
        }
    }
    return 0;
}