树链剖分-SPOJ375(QTREE)

QTREE - Query on a tree

You are given a tree (an acyclic undirected connected graph) with N nodes, and edges numbered 1, 2, 3...N-1.

We will ask you to perfrom some instructions of the following form:

• CHANGE i ti : change the cost of the i-th edge to ti
  or
• QUERY a b : ask for the maximum edge cost on the path from node a to node b

Input

The first line of input contains an integer t, the number of test cases (t <= 20). t test cases follow.

For each test case:

• In the first line there is an integer N (N <= 10000),
• In the next N-1 lines, the i-th line describes the i-th edge: a line with three integers a b c denotes an edge between a, b of cost c (c <= 1000000),
• The next lines contain instructions "CHANGE i ti" or "QUERY a b",
• The end of each test case is signified by the string "DONE".

There is one blank line between successive tests.

Output

For each "QUERY" operation, write one integer representing its result.

Example

Input:
1

3
1 2 1
2 3 2
QUERY 1 2
CHANGE 1 3
QUERY 1 2
DONE

Output:
1
3

传送门
一句话题意：给出一棵树，要求在树上：
　　1）更新某个节点的值
　　2）求两个节点间最短路径上的最大值

模板题，数据结构解析详见代码。

 #include<cstring>
 #include<cstdio>
 #include<algorithm>
 #include<cmath>
 #define clr(a) memset(a,0,sizeof(a));
 #define foru(i,x,y) for(int i=x;i<=y;i++)
 using namespace std;
 const int N=;
 struct edge{int to,nxt;}e[N*];
 int t[N],a[N][],d[N],id[N],head[N],f[N],siz[N],son[N],top[N];
 //f[v]        节点v的父节点编号
 //id[v]        节点v的父边在线段树中的编号
 //siz[v]    以节点v为根的数中的节点数
 //son[v]    节点v的子节点中siz[]最大的节点编号
 //top[v]    节点v所在重链的顶端节点编号
 //d[v]        节点v的深度
 int ne,cnt,n;

 void add(int a,int b){
     e[++ne]=(edge){b,head[a]};head[a]=ne;
 }

 /////////////////////////////树剖////////////////////////////////////

 void dfs(int k,int fa,int dep){//统计f[] siz[] son[] d[]
     f[k]=fa;d[k]=dep;siz[k]=;son[k]=;
     for(int i=head[k];i;i=e[i].nxt){
         int v=e[i].to;
         if(v==fa)continue;//无向图判是否走过，易漏
         dfs(v,k,dep+);
         siz[k]+=siz[v];
         if(siz[v]>siz[son[k]])son[k]=v;
     }
 }

 void build(int k,int tp){
     id[k]=++cnt;//按序将边加入线段树
     top[k]=tp;
     if(son[k])build(son[k],tp);//重儿子的top[]从重链顶端继承
     for(int i=head[k];i;i=e[i].nxt)
         if(e[i].to!=son[k]&&e[i].to!=f[k])
             build(e[i].to,e[i].to);//轻儿子top[]为自身
 }
 /////////////////////////////树剖////////////////////////////////////

 /////////////////////////////线段树////////////////////////////////////
 #define mid ((L+R)>>1)
 #define ls (k<<1)//写位运算一定要开-Wall，否则一定要记得加括号
 #define rs ls+1

 void update(int k,int L,int R,int p,int x){
     if(p>R||p<L)return;
     if(L==R){t[k]=x;return;}
     update(ls,L,mid,p,x); update(rs,mid+,R,p,x);
     t[k]=max(t[ls],t[rs]);
 }

 int query(int k,int L,int R,int l,int r){
     if(l>R||r<L)return ;
     if(l<=L&&R<=r)return t[k];
     return max(query(ls,L,mid,l,r),query(rs,mid+,R,l,r));
 }
 /////////////////////////////线段树////////////////////////////////////

 int find(int x,int y){
     int ans=-1e9;
     while(top[x]!=top[y]){//类似LCA，每次将较低的节点上跳，并统计路径上的最大值
         if(d[top[x]]<d[top[y]])swap(x,y);
         ans=max(ans,query(,,cnt,id[top[x]],id[x]));
         x=f[top[x]];
     }
     if(d[x]>d[y])swap(x,y);//当两点处于同一条链上的时候，进行最后一次统计
     if(x!=y)ans=max(ans,query(,,cnt,id[x]+,id[y]));//注意，因为id[x]为x的父边，所以若不+1，就会多统计一条边
     return ans;
 }

 char ch[];
 void work(){
     int x,y;
     while(scanf("%s",ch),ch[]!='D'){
         scanf("%d%d",&x,&y);
         if(ch[]=='C')update(,,cnt,id[a[x][]],y);
         else printf("%d\n",find(x,y));
     }
 }

 void init(){
     clr(a);clr(e);clr(t);clr(son);clr(siz);clr(id);clr(top);clr(head);clr(d);clr(f);
     ne=cnt=n=;
     scanf("%d",&n);
     foru(i,,n-){
         scanf("%d%d%d",&a[i][],&a[i][],&a[i][]);
         add(a[i][],a[i][]);add(a[i][],a[i][]);
     }
     dfs(,,);
     build(,);
     foru(i,,n-){
         if(d[a[i][]]>d[a[i][]])swap(a[i][],a[i][]);
         update(,,cnt,id[a[i][]],a[i][]);//建树
     }
 }

 int main(){
     int T;
     scanf("%d",&T);
     while(T--){
         init();
         work();
     }
 }