http://poj.org/problem?id=2763
树链剖分 (边权的计算)
每个儿子对应一个父亲,那么就把边权放在儿子点的位置上。在lca的部分计算时候注意一下。
#include <cstdio>
#include <vector>
using namespace std;
typedef long long ll;
const int N=100005;
int n,q,s,cnt=0;
int father[N],depth[N],hson[N],size[N],top[N],dfn[N],rev[N],btm[N];//btm该点下最大的dfn
//x的父亲 x的深度 x的重儿子 x下结点数 x重链顶部 seg(线段树中的下标||dfn) rev[seg[x]]=x得到原编号
vector<pair<int,int> >v[N];
int num[N];
int sum[N*4];
int add[N*4];
void dfs1(int san,int fa){//处理father depth hson size数组
depth[san]=depth[fa]+1;
size[san]=1;
father[san]=fa;
for(int i=0;i<v[san].size();i++){
if(v[san][i].first==fa)continue;
dfs1(v[san][i].first,san);
size[san]+=size[v[san][i].first];//更新size
if(size[v[san][i].first]>size[hson[san]]){//判断这条链是否超过之前的重儿子
hson[san]=v[san][i].first;
}
}
}
void dfs2(int san){//printf("%d",dfn[0]);
if(hson[san]){
dfn[hson[san]]=++dfn[0];
top[hson[san]]=top[san];
rev[dfn[0]]=hson[san];
dfs2(hson[san]);
}
for(int i=0;i<v[san].size();i++){
if(top[v[san][i].first])continue;
dfn[v[san][i].first]=++dfn[0];
rev[dfn[0]]=v[san][i].first;
top[v[san][i].first]=v[san][i].first;
dfs2(v[san][i].first);
}
}
int query(int k,int l,int r,int x,int y){
if(l>=x&&r<=y)return sum[k];
int mid=l+r>>1;
int res=0;
if(x<=mid)res+=query(k<<1,l,mid,x,y);
if(y>mid)res+=query(k<<1|1,mid+1,r,x,y);
return res;
}
int ask(int x,int y){
int fx=top[x],fy=top[y];int ans=0;
while(fx!=fy){
if(depth[fx]<depth[fy])swap(x,y),swap(fx,fy);
ans+=query(1,1,dfn[0],dfn[fx],dfn[x]);//fx的顶部所以dfn小要放在前面
x=father[fx];
fx=top[x];
}
if(depth[x]>depth[y])swap(x,y);
if(hson[x]) ans += query(1,1,dfn[0],dfn[hson[x]],dfn[y]);
return ans;
}
void build(int k,int l,int r){
int mid=l+r>>1;
if(l==r){
sum[k]=num[l];//num是线段树的最底层
return ;
}
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
sum[k]=sum[k<<1]+sum[k<<1|1];
}
void update(int k,int kk,int l,int r,int v){
int mid=l+r>>1;
if(r<kk||l>kk)return;
if(l==r&&l==kk){
sum[k]=v;
return ;
}
update(k<<1,kk,l,mid,v);
update(k<<1|1,kk,mid+1,r,v);
sum[k]=sum[k<<1]+sum[k<<1|1];
}
struct node{
int xx,yy,ll;
}WZ[N];
int main(){
scanf("%d%d%d",&n,&q,&s);
for(int i=1;i<n;i++){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
v[x].push_back(make_pair(y,z));
v[y].push_back(make_pair(x,z));
WZ[i].xx=x;
WZ[i].ll=z;
WZ[i].yy=y;
}
dfn[0]=dfn[1]=top[1]=top[1]=1;
dfs1(1,0);
dfs2(1);
for(int i=1;i<n;i++){
if(depth[WZ[i].xx]>depth[WZ[i].yy]){
swap(WZ[i].xx,WZ[i].yy);
}
num[dfn[WZ[i].yy]]=WZ[i].ll;//这里是给线段树的最下层赋值
}
build(1,1,dfn[0]);
while(q--){
int op,a1,a2;
scanf("%d",&op);
if(op){
scanf("%d%d",&a1,&a2);
update(1,dfn[WZ[a1].yy],1,dfn[0],a2);//
}
else{
scanf("%d",&a1);
printf("%d\n",ask(s,a1));
s=a1;
}
}
}