POJ2763Housewife Wind

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;
		}
	}
}