BZOJ3083: 遥远的国度【树链剖分】

题目描述：

问题是这样的：遥远的国度有n个城市，这些城市构成了一颗树。
这个国度有一个首都，我们可以把这个首都看做整棵树的根，首都是随时有可能变为另外一个城市的。
遥远的国度的每个城市有一个防御值，有些时候RapiD会使得某两个城市之间的路径上的所有城市的防御值都变为某个值。
RapiD想知道在某个时候，如果把首都看做整棵树的根的话，那么以某个城市为根的子树的所有城市的防御值最小是多少。
由于RapiD无法解决这个问题，所以他拦住了zcwwzdjn希望他能帮忙。但zcwwzdjn还要追杀sb的zhx，所以这个重大的问题就被转交到了你的手上。
$n\le 100000$

题目分析：

以1为根树链剖分+线段树进行链修改和查询，查询的时候如果$root=x$则查询整颗子树；$root$在$x$的子树中则返回整棵树除$x$到$root$方向的儿子这一支子树；否则直接返回$x$的子树。

写这个blog主要是记录这个用树剖找链上祖先的下一个节点的代码：

int find(int t,int u){
    for(;top[t]!=top[u];u=fa[top[u]]) if(fa[top[u]]==t) return top[u];
    return son[t];
}

Code：

#include<cstdio>
#include<vector>
#include<algorithm>
#define maxn 100005
using namespace std;
const int inf = 2147483647;
int n,m,a[maxn],fa[maxn],dep[maxn],son[maxn],top[maxn],siz[maxn],dfn[maxn],ln[maxn],tim;
vector<int>G[maxn];
void dfs1(int u,int ff){
    dep[u]=dep[fa[u]=ff]+1,siz[u]=1;
    for(int i=G[u].size()-1,v;i>=0;i--) if((v=G[u][i])!=ff){
        dfs1(v,u),siz[u]+=siz[v];
        if(siz[v]>siz[son[u]]) son[u]=v;
    }
}
void dfs2(int u,int tp){
    top[u]=tp,ln[dfn[u]=++tim]=u;
    if(son[u]) dfs2(son[u],tp);
    for(int i=G[u].size()-1,v;i>=0;i--) if(!dfn[v=G[u][i]]) dfs2(v,v);
}
int mn[maxn<<2],tag[maxn<<2];
void build(int i,int l,int r){
    tag[i]=-1;
    if(l==r) {mn[i]=a[ln[l]];return;}
    int mid=(l+r)>>1;
    build(i<<1,l,mid),build(i<<1|1,mid+1,r);
    mn[i]=min(mn[i<<1],mn[i<<1|1]);
}
inline void pushdown(int i){
    if(~tag[i]) mn[i<<1]=tag[i<<1]=mn[i<<1|1]=tag[i<<1|1]=tag[i],tag[i]=-1;
}
void insert(int i,int l,int r,int x,int y,int d){
    if(x<=l&&r<=y) {mn[i]=tag[i]=d;return;}
    int mid=(l+r)>>1;
    pushdown(i);
    if(x<=mid) insert(i<<1,l,mid,x,y,d);
    if(y>mid) insert(i<<1|1,mid+1,r,x,y,d);
    mn[i]=min(mn[i<<1],mn[i<<1|1]);
}
int query(int i,int l,int r,int x,int y){
    if(x<=l&&r<=y) return mn[i];
    int mid=(l+r)>>1,ret=inf;
    pushdown(i);
    if(x<=mid) ret=min(ret,query(i<<1,l,mid,x,y));
    if(y>mid) ret=min(ret,query(i<<1|1,mid+1,r,x,y));
    return ret;
}
int LCA(int u,int v){
    for(;top[u]!=top[v];u=fa[top[u]]) if(dep[top[u]]<dep[top[v]]) swap(u,v);
    return dep[u]<dep[v]?u:v;
}
void modify(int u,int v,int d){
    for(;top[u]!=top[v];u=fa[top[u]]){
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        insert(1,1,n,dfn[top[u]],dfn[u],d);
    }
    if(dep[u]<dep[v]) swap(u,v);
    insert(1,1,n,dfn[v],dfn[u],d);
}
int find(int t,int u){
    for(;top[t]!=top[u];u=fa[top[u]]) if(fa[top[u]]==t) return top[u];
    return son[t];
}
int main()
{
    int op,x,y,z,rt;
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++) scanf("%d%d",&x,&y),G[x].push_back(y),G[y].push_back(x);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    scanf("%d",&rt);
    dfs1(1,0),dfs2(1,1);
    build(1,1,n);
    while(m--){
        scanf("%d",&op);
        if(op==1) scanf("%d",&rt);
        if(op==2) scanf("%d%d%d",&x,&y,&z),modify(x,y,z);
        if(op==3){
            scanf("%d",&x);
            if(x==rt) printf("%d\n",mn[1]);
            else if(LCA(x,rt)!=x) printf("%d\n",query(1,1,n,dfn[x],dfn[x]+siz[x]-1));
            else y=find(x,rt),printf("%d\n",min(query(1,1,n,1,dfn[y]-1),dfn[y]+siz[y]<=n?query(1,1,n,dfn[y]+siz[y],n):inf));
        }
    }
}