SPOJ - QTREE Query on a tree 树链剖分 基于边权

题目链接：点击查看

题意：给出n个点，每条边有边权，两种操作，1、查询路径上最大的权值  2、修改某边的权值

题解：把每条边赋给儿子，然后就是基本操作了。要注意的是，修改边的时候给出的是第几条边，因此输入的时候要保存一下

#include <bits/stdc++.h>
using namespace std;
const int N=1e4+10;
struct edge{
	int to,nex,d;
}e[N*2];
int head[N],len;
int fa[N],deep[N],son[N],num[N],top[N],val[N];
int id[N],pid[N],cnt;
int posl[N],posr[N],posval[N];
int n;
struct node{
	int l,r;
	int val;
}tree[N<<2];
void init()
{
	for(int i=1;i<=n;i++)
	{
		head[i]=-1;
		son[i]=-1;
		fa[i]=-1;
	}
	cnt=0;
	len=0;
}
void addedge(int u,int v,int d)
{
	e[len].to=v;
	e[len].d=d;
	e[len].nex=head[u];
	head[u]=len++;
}
void dfs1(int u,int f)
{
	num[u]=1;
	int to;
	for(int i=head[u];i!=-1;i=e[i].nex)
	{
		to=e[i].to;
		if(to==f) continue;
		val[to]=e[i].d;
		fa[to]=u;
		deep[to]=deep[u]+1;
		dfs1(to,u);
		if(son[u]==-1||num[to]>num[son[u]])
			son[u]=to;
	}
}
void dfs2(int u,int topfa)
{
	top[u]=topfa;
	id[u]=++cnt; pid[cnt]=u;
	if(son[u]==-1) return;
	dfs2(son[u],topfa);
	int to;
	for(int i=head[u];i!=-1;i=e[i].nex)
	{
		to=e[i].to;
		if(to==fa[u]||to==son[u]) continue;
		dfs2(to,to);
	}
}
void pushup(int cur)
{
	tree[cur].val=max(tree[cur<<1].val,tree[cur<<1|1].val);
}
void build(int l,int r,int cur)
{
	tree[cur].l=l;
	tree[cur].r=r;
	if(l==r)
	{
		tree[cur].val=val[pid[l]];
		return;
	}
	int mid=(r+l)>>1;
	build(l,mid,cur<<1);
	build(mid+1,r,cur<<1|1);
	pushup(cur);
}
void update(int pos,int cur,int val)
{
	if(tree[cur].l==tree[cur].r)
	{
		tree[cur].val=val;
		return;
	}
	if(pos<=tree[cur<<1].r) update(pos,cur<<1,val);
	else update(pos,cur<<1|1,val);
	pushup(cur);
//	cout<<tree[cur].val<<endl;
}
int query(int pl,int pr,int cur)
{
	if(pl<=tree[cur].l&&tree[cur].r<=pr) return tree[cur].val;
	int res=-1e9;
	if(pl<=tree[cur<<1].r) res=max(res,query(pl,pr,cur<<1));
	if(pr>=tree[cur<<1|1].l) res=max(res,query(pl,pr,cur<<1|1));
	return res;
}
int solve(int l,int r)
{
	int f1=top[l],f2=top[r];
	int res=-1e9;
	while(f1!=f2)
	{
		if(deep[f1]<deep[f2])
		{
			swap(f1,f2);
			swap(l,r);
		}
		res=max(res,query(id[f1],id[l],1));
		l=fa[f1];
		f1=top[l];
	}
	if(deep[l]>deep[r]) swap(l,r);
//	cout<<id[l]<<" "<<id[r]<<endl;
	res=max(res,query(id[l]+1,id[r],1));
	return res;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		init();
		int u,v,d;
		char op[10];
		for(int i=1;i<n;i++)
		{
			scanf("%d%d%d",&u,&v,&d);
			posl[i]=u,posr[i]=v,posval[i]=d;
			addedge(u,v,d);
			addedge(v,u,d);
		}
		dfs1(1,-1);
		dfs2(1,1);
		build(1,n,1);
		
		while(scanf("%s",op)&&strcmp(op,"DONE")!=0)
		{
			scanf("%d%d",&u,&v);
			if(op[0]=='Q') printf("%d\n",solve(u,v));
			else {
				if(fa[posl[u]]==posr[u]) u=posl[u];
				else u=posr[u];
				 update(id[u],1,v);
			}
		}
	}
	return 0;	
}