poj2763 Housewife Wind【树链剖分+线段树】

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <stdlib.h>
#include <time.h>
using namespace std;
typedef vector<int> VI;
typedef double db;
const int N=100000+10;
const int inf=0x7fffffff;
struct node
{
int to,next;
}g[2*N];
int cnt,fa
,dep
,sz
,son
,w
,top
,tot,head
,mp
[3];

void add_Node(int x,int y)
{
g[cnt].to=y;
g[cnt].next=head[x];
head[x]=cnt++;
}

void dfs1(int x,int f)
{
int y,mx=-1;
fa[x]=f;
sz[x]=1;
dep[x]=dep[f]+1;
for(int i=head[x];i!=-1;i=g[i].next)
{
y=g[i].to;
if(y==f) continue;
dfs1(y,x);
sz[x]+=sz[y];
if(sz[y]>mx){mx=sz[y];son[x]=y;}
}
}

void dfs2(int x,int tp)
{
top[x]=tp;
if(son[x]!=0)
{
w[x]=tot++;
dfs2(son[x],tp);
}
else
{
w[x]=tot++;
return ;
}
int y;
for(int i=head[x];i!=-1;i=g[i].next)
{
y=g[i].to;
if(y!=son[x]&&y!=fa[x])
dfs2(y,y);
}
}

struct Tree
{
int l,r,mx;
}tree[4*N];
void build(int o,int l,int r)
{
tree[o].l=l;
tree[o].r=r;
tree[o].mx=0;
if(l!=r)
{
int mid=(l+r)/2;
build(o*2,l,mid);
build(o*2+1,mid+1,r);
}
}

void update(int o,int l,int r,__int64 val)
{
if(l<=tree[o].l&&tree[o].r<=r)
{
tree[o].mx=val;
}
else
{
int mid=(tree[o].l+tree[o].r)/2;
if(l<=mid)  update(o*2,l,r,val);
if(r>mid) update(o*2+1,l,r,val);
tree[o].mx=tree[o*2].mx+tree[o*2+1].mx;
}
}

__int64 query(int o,int l,int r)
{
if(l<=tree[o].l&&tree[o].r<=r)
{
return tree[o].mx;
}
else
{
__int64 ans=0;
int mid=(tree[o].l+tree[o].r)/2;
if(l<=mid) ans+=query(o*2,l,r);
if(r>mid) ans+=query(o*2+1,l,r);
return ans;
}
}

__int64 Find(int x,int y)
{
int f1=top[x],f2=top[y];
__int64 tmp=0;
while(f1!=f2)
{
if(dep[f1]<dep[f2])
{
swap(f1,f2);
swap(x,y);
}
tmp+=query(1,w[f1],w[x]);
x=fa[f1];
f1=top[x];
}
if(x==y) return tmp;
if(dep[x]>dep[y]) swap(x,y);
return tmp+query(1,w[son[x]],w[y]);
}

int main()
{
int _,n,a,b,c,m,s;
__int64 t;
while(~scanf("%d%d%d",&n,&m,&s))
{
cnt=0;
memset(head,0xff,sizeof(head));
for(int i=0;i<n-1;i++)
{
scanf("%d%d%I64d",&a,&b,&t);
mp[i][0]=a;
mp[i][1]=b;
mp[i][2]=t;
add_Node(a,b);
add_Node(b,a);
}
memset(son,0,sizeof(son));
tot=1;
dep[0]=0;
dfs1(1,0);
dfs2(1,1);
build(1,1,tot-1);
for(int i=0;i<n-1;i++)
{
a=mp[i][0];b=mp[i][1];
if(dep[a]>dep[b])
swap(mp[i][1],mp[i][0]);
update(1,w[mp[i][1]],w[mp[i][1]],mp[i][2]);
}
for(int i=0;i<m;i++)
{
scanf("%d",&c);
if(c==1)
{
scanf("%d%I64d",&a,&t);
update(1,w[mp[a-1][1]],w[mp[a-1][1]],t);
}
else
{
scanf("%d",&b);
printf("%I64d\n",Find(s,b));
s=b;
}
}
}
return 0;
}