FZU - 2082 过路费.轻重链剖分

连接：https://vjudge.net/problem/FZU-2082

用轻重链剖分和线段树维护查询就好，注意除了线段树查询外不要开longlong，不然可能会re，可能测评机比较奇葩吧

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <memory.h>
using namespace std;
typedef long long ll;
const ll maxn=1e5+10;
#define mid ((l+r)/2)
struct E{
	int v,w,next;
};

struct D{
	int u,v,w;
};
int fa[maxn],tree[maxn],dep[maxn],son[maxn],size[maxn],top[maxn];
int head[maxn];
E edge[maxn];
D nmsl[maxn];
int lenid=0;
int tot=1;
void add(int u,int v,int w){
	edge[tot].w=w;
	edge[tot].v=v;
	edge[tot].next=head[u];
	head[u]=tot++;
}

void dfs1(int u,int f,int d){
	dep[u]=d;
	size[u]=1;
	son[u]=0;
	fa[u]=f;
	for(int i=head[u];~i;i=edge[i].next){
		int v=edge[i].v;
		if(v!=f){
			dfs1(v,u,d+1);
			size[u]+=size[v];
			if(son[u]==0 || size[son[u]]<size[v]){
				son[u]=v;
			}
		}
	}
}

void dfs2(int u,int f){
	tree[u]=++lenid;
	top[u]=f;
	if(son[u]){
		dfs2(son[u],f);
	}
	for(int i=head[u];~i;i=edge[i].next){
		int v=edge[i].v;
		if(v!=fa[u] && v!=son[u]){
			dfs2(v,v);
		}
	}
}

int nums[maxn];
ll sum[maxn<<2];
void pushup(int rt){
	sum[rt]=sum[rt<<1|1]+sum[rt<<1];
}

void build(int l,int r,int rt){
	if(l==r){
		sum[rt]=nums[l];
		return ;
	}
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	pushup(rt);
}

void update(int x,int k,int l,int r,int rt){
	if(l==r){
		sum[rt]=k;
		return ;
	}
	if(x<=mid){
		update(x,k,l,mid,rt<<1);
	}else{
		update(x,k,mid+1,r,rt<<1|1);
	}
	pushup(rt);
}

ll query(int a,int b,int l,int r,int rt){
	if(a<=l && b>=r){
		return sum[rt];
	}
	ll ans=0;
	if(a<=mid){
		ans+=query(a,b,l,mid,rt<<1);
	}
	if(b>mid){
		ans+=query(a,b,mid+1,r,rt<<1|1);
	}
	pushup(rt);
	return ans;
}

ll lca(int x,int y){
	ll ans=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]){
			swap(x,y);
		}
		ans+=query(tree[top[x]],tree[x],1,lenid,1);
		x=fa[top[x]];
	}
	if(x==y) return ans;
	if(dep[x]>dep[y]){
		swap(x,y);
	}
	ans+=query(tree[son[x]],tree[y],1,lenid,1);
	return ans;
}

void init(){
	lenid=0;
	tot=0;
	memset(head,-1,sizeof(head));
	memset(son,0,sizeof(son));
}

int main(){
	int n,m;
	while(~scanf("%d%d",&n,&m)){
		init();
		for(int i=1;i<=n-1;i++){
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			nmsl[i].u=u;
			nmsl[i].v=v;
			nmsl[i].w=w;
			add(u,v,w);
			add(v,u,w);
		}
		dfs1(1,0,1);
		dfs2(1,1);
		for(int i=1;i<n;i++){
			if(dep[nmsl[i].u]<dep[nmsl[i].v]){
				swap(nmsl[i].u,nmsl[i].v);
			}
			nums[tree[nmsl[i].u]]=nmsl[i].w;
		}
		build(1,n,1);
		while(m--){
			int op,x,y;
			scanf("%d%d%d",&op,&x,&y);
			if(op){
				printf("%lld\n",lca(x,y));
			}else{
				update(tree[nmsl[x].u],y,1,n,1);
			}
		}
	}
	return 0;
}