基于边的树链剖分,与点的几个不同点:
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;
}