poj 2763 Housewife Wind （树链剖分）

题目链接：http://poj.org/problem?id=2763

题意：

给定一棵含n个结点的树和树的边权，共有q次操作，分为两种

0 c ：求从位置s到c的距离，然后s变成c

1 a b:把第a条边的权值变为b

分析：

树链剖分，注意查询后要改变起点

代码如下：

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<cstring>

 using namespace std;
 const int maxn=;

 struct Edge
 {
     int to,next;
 }edge[maxn*];
 int head[maxn];
 int cnt,tmp,n;
 int dep[maxn],fa[maxn],size[maxn],son[maxn],top[maxn],id[maxn],rank[maxn];
 struct Node
 {
     int u,v,c;
 }node[maxn];

 void init()
 {
     memset(head,-,sizeof(head));
     memset(son,-,sizeof(son));
     tmp=;
     cnt=;
 }

 void addedge(int u,int v)
 {
     edge[cnt].to=v;
     edge[cnt].next=head[u];
     head[u]=cnt++;
 }

 void dfs_1(int u,int f,int d)
 {
     dep[u]=d;
     size[u]=;
     fa[u]=f;
     for(int i=head[u];i!=-;i=edge[i].next)
     {
         int v=edge[i].to;
         if(v==f)
             continue;
         dfs_1(v,u,d+);
         size[u]+=size[v];
         if(son[u]==-||size[son[u]]<size[v])
             son[u]=v;
     }
 }

 void dfs_2(int u,int tp)
 {
     top[u]=tp;
     id[u]=++tmp;
     rank[tmp]=u;
     if(son[u]!=-)
         dfs_2(son[u],tp);
     for(int i=head[u];i!=-;i=edge[i].next)
     {
         int v=edge[i].to;
         if(v!=fa[u]&&v!=son[u])
             dfs_2(v,v);
     }
 }
 struct Tree
 {
     int left,right;
     int t;
 }tree[maxn*];

 void pushup(int i)
 {
     tree[i].t=tree[i*].t+tree[i*+].t;
 }

 void build(int i,int begin,int end)
 {
     tree[i].left=begin;
     tree[i].right=end;
     tree[i].t=;
     if(begin==end)
         return;
     int mid=(begin+end)/;
     build(i*,begin,mid);
     build(i*+,mid+,end);
 }

 void update(int i,int k,int val)
 {
     if(tree[i].left==k&&tree[i].right==k)
     {
         tree[i].t=val;
         return;
     }
     int mid=(tree[i].left+tree[i].right)/;
     if(k<=mid)
         update(i*,k,val);
     else
         update(i*+,k,val);
     pushup(i);
 }

 int query(int i,int begin,int end)
 {
     if(tree[i].left>=begin&&tree[i].right<=end)
         return tree[i].t;
     int mid=(tree[i].left+tree[i].right)/;
     int res=;
     if(mid>=begin)
         res+=query(i*,begin,end);
     if(mid<end)
         res+=query(i*+,begin,end);
     return res;
 }

 int find(int u,int v)
 {
     int tp1=top[u],tp2=top[v];
     int res=;
     while(tp1!=tp2)
     {
         if(dep[tp1]<dep[tp2])
         {
             swap(tp1,tp2);
             swap(u,v);
         }
         res+=query(,id[tp1],id[u]);
         u=fa[tp1];
         tp1=top[u];
     }
     if(u==v)
         return res;
     if(dep[u]>dep[v])
         swap(u,v);
     res+=query(,id[son[u]],id[v]);
     return res;
 }

 int main()
 {
     int n,q,s;
     scanf("%d%d%d",&n,&q,&s);
     init();
     for(int i=;i<n;i++)
     {
         scanf("%d%d%d",&node[i].u,&node[i].v,&node[i].c);
         addedge(node[i].u,node[i].v);
         addedge(node[i].v,node[i].u);
     }
     dfs_1(,,);
     dfs_2(,);
     build(,,tmp);
     for(int i=;i<n;i++)
     {
         if(dep[node[i].u]>dep[node[i].v])
             swap(node[i].u,node[i].v);
         update(,id[node[i].v],node[i].c);
     }
     for(int i=;i<q;i++)
     {
         int a,k,w;
         scanf("%d",&a);
         if(a==)
         {
             scanf("%d",&k);
             printf("%d\n",find(s,k));
             s=k;
         }
         else
         {
             scanf("%d%d",&k,&w);
             update(,id[node[k].v],w);
         }
     }
     return ;
 }