SPOJ 375. Query on a tree （基于边的树链剖分）

基于边的树链剖分，与点的几个不同点：

1，维护边权值，即是维护某点与其父亲点之间边的权值，所以可以等同视为该点的权值（即孩子结点，可以通过判断深度较大的是孩子结点）。

2，在LCA过程中，最后一步，在重链上求区间最大值，这里的左端点实际上是对应一条不在计算范围内的边，所以调整左端点x为son[x]。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cmath>
#include<map>
#include<string>
#include<set>
#include<queue>
#include<algorithm>
using namespace std;
const int inf=0x80000000;
const int maxn=10010;
int head[maxn],to[maxn*2],nxt[maxn*2],edge;
int siz[maxn],dep[maxn],son[maxn],fa[maxn],top[maxn],tot;
int id[maxn],rid[maxn];
int n;
int e[maxn][3];
int maxi[maxn*4];
void init()
{
    tot=0;
    memset(son,-1,sizeof(son));
    edge=0;
    memset(head,-1,sizeof(head));
}
void addEdge(int u,int v)
{
    to[edge]=u,nxt[edge]=head[v],head[v]=edge++;
    to[edge]=v,nxt[edge]=head[u],head[u]=edge++;
}
void dfs1(int now,int f,int d)
{
    siz[now]=1;
    dep[now]=d;
    fa[now]=f;
    for(int i=head[now]; i!=-1; i=nxt[i])
    {
        int v=to[i];
        if(v!=f)
        {
            dfs1(v,now,d+1);
            siz[now]+=siz[v];
            if(son[now]==-1||siz[son[now]]<siz[v])
                son[now]=v;
        }
    }
}
void dfs2(int now,int tp)
{
    id[now]=++tot;
    rid[id[now]]=now;
    top[now]=tp;
    if(son[now]==-1)
        return ;
    dfs2(son[now],tp);
    for(int i=head[now]; i!=-1; i=nxt[i])
    {
        int v=to[i];
        if(v!=fa[now]&&v!=son[now])
        {
            dfs2(v,v);
        }
    }
}
void push_up(int o)
{
    maxi[o]=max(maxi[o<<1],maxi[o<<1|1]);
}
void build(int o,int l,int r)
{
    if(l==r)
    {
        maxi[o]=0;
    }
    else
    {
        int m=l+(r-l)/2;
        build(o<<1,l,m);
        build(o<<1|1,m+1,r);
        push_up(o);
    }
}
void update(int o,int L,int R,int pos,int val)
{
    if(L==R)
        maxi[o]=val;
    else
    {
        int M=L+(R-L)/2;
        if(pos<=M)
            update(o<<1,L,M,pos,val);
        else
            update(o<<1|1,M+1,R,pos,val);
        push_up(o);
    }
}
int queryMax(int o,int L,int R,int ql,int qr)
{
    if(ql<=L&&R<=qr)
        return maxi[o];
    else
    {
        int M=L+(R-L)/2;
        int res=inf;
        if(ql<=M)
            res=max(res,queryMax(o<<1,L,M,ql,qr));
        if(M<qr)
            res=max(res,queryMax(o<<1|1,M+1,R,ql,qr));
        return res;
    }
}
int getMax(int x,int y)
{
    int res=inf;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]>dep[top[y]])
            swap(x,y);
        res=max(res,queryMax(1,1,n,id[top[y]],id[y]));
        y=fa[top[y]];
    }
    if(dep[x]>dep[y])
        swap(x,y);
    res=max(res,queryMax(1,1,n,id[son[x]],id[y]));
    return res;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d",&n);
        for(int i=1; i<n; ++i)
        {
            scanf("%d%d%d",&e[i][0],&e[i][1],&e[i][2]);
            addEdge(e[i][0],e[i][1]);
        }
        dfs1(1,-1,1);
        dfs2(1,1);
        build(1,1,n);
        for(int i=1;i<n;++i)
        {
            if(dep[e[i][0]]>dep[e[i][1]])
                swap(e[i][0],e[i][1]);
            update(1,1,n,id[e[i][1]],e[i][2]);
        }
        char str[10];
        while(scanf("%s",str)&&str[0]!='D')
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(str[0]=='Q')
                printf("%d\n",getMax(a,b));
            else
                update(1,1,n,id[e[a][1]],b);
        }
        if(T)
            printf("\n");
    }
    return 0;
}