树链剖分

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
struct node
{
    int x,y,v;
};
node edge[10001];
char c[10];
int t,n,z,m,a,b,f1,f2,root,w[10001],tree[40004],top[10001],size[10001],son[10001],head[10001],next[20002],list[20002],fa[10001],dep[10001]; 
//tree[]为线段树，size[v]表示以v为根节点的子树的结点个数，son[v]表示v的重儿子，fa[v]表示v的父节点，dep[v]表示树的深度，top[v]表示v所在链的顶端节点，w[v]表示点v与其父亲的连边在线段树中的位置。
//n表示节点数，m表示边数，z表示线段树中的边数。
void insert(int x,int y)
{
    next[++m]=head[x];
    head[x]=m;
    list[m]=y;
}
void dfs(int v) //dfs求节点的size son dep 
{
    size[v]=1;
    son[v]=0;
    for (int i=head[v];i;i=next[i])
        if (list[i]!=fa[v])
        {
            fa[list[i]]=v;
            dep[list[i]]=dep[v]+1;
            dfs(list[i]);
            if (size[list[i]]>size[son[v]]) son[v]=list[i];
            size[v]+=size[list[i]];
        }
}
void build_tree(int v,int tp) //求top w
{
    w[v]=++z;
    top[v]=tp;
    if (son[v]!=0) build_tree(son[v],top[v]);
    for (int i=head[v];i;i=next[i])
        if (list[i]!=son[v]&&list[i]!=fa[v])
            build_tree(list[i],list[i]);
}
void updata(int root,int low,int high,int pos,int x) //更改某条边（即线段树中第pos个点）的值
{
    if (pos>high||pos<low) return;
    if (low==high)
    {
        tree[root]=x;
        return;
    }
    int mid=(low+high)/2,ls=root*2,rs=ls+1;
    updata(ls,low,mid,pos,x);
    updata(rs,mid+1,high,pos,x);
    tree[root]=max(tree[ls],tree[rs]);
}
int maxi(int root,int low,int high,int l,int r) 求区间[l,r]的最大值，线段树。
{
    if (l>high||r<low) return 0;
    if (l<=low&&high<=r) return tree[root];
    int mid=(low+high)/2,ls=root*2,rs=ls+1;
    return max(maxi(ls,low,mid,l,r),maxi(rs,mid+1,high,l,r));
}
int find(int va,int vb) 寻找va->vb路径上边的最大值，线段树维护。
{
    int f1=top[va],f2=top[vb],tmp=0;
    while (f1!=f2)
    {
        if (dep[f1]<dep[f2])
        {
            swap(f1,f2);
            swap(va,vb);
        }
        tmp=max(tmp,maxi(1,1,z,w[f1],w[va]));
        va=fa[f1];f1=top[va];
    }
    if (va==vb) return tmp;
    if (dep[va]>dep[vb]) swap(va,vb);
    return max(tmp,maxi(1,1,z,w[son[va]],w[vb]));
}        
int main()
{
    scanf("%d",&t);
    while (t--)
    {
        memset(size,0,sizeof(size));
        memset(head,0,sizeof(head));
        memset(next,0,sizeof(next));
        memset(tree,0,sizeof(tree));
        scanf("%d",&n);
        root=(n+1)/2;
        fa[root]=z=dep[root]=m=0;
        for (int i=1;i<n;i++)
        {
            scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].v);
            insert(edge[i].x,edge[i].y);
            insert(edge[i].y,edge[i].x);
        }
        dfs(root);
        build_tree(root,root);
        for (int i=1;i<n;i++)
        {
            if (dep[edge[i].x]>dep[edge[i].y]) swap(edge[i].x,edge[i].y);
            updata(1,1,z,w[edge[i].y],edge[i].v);
        }
    }
    scanf("%s",c);
    while (c[0]!=‘D‘)
    {
        scanf("%d%d",&a,&b);
        if (c[0]==‘Q‘) printf("%d\n",find(a,b));
                    else updata(1,1,z,w[edge[a].y],b);
        scanf("%s",c);
    }
    return 0;
}