BZOJ 1036 树的统计

Description

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 III. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

Input

输入的第一行为一个整数n，表示节点的个数。接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。接下来1行，为一个整数q，表示操作的总数。接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

Output

对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

Sample Input

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output

4
1
2
2
10
6
5
6
5
16

HINT

 

Source

 树链剖分大裸题。

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 using namespace std;
 #define maxn 30010
 #define root 1
 int n,next[*maxn],side[maxn],toit[*maxn],key[maxn],tot,son[maxn],father[maxn];
 int edge[maxn],pos[maxn],top[maxn],dep[maxn],heavy[maxn],ord[maxn];
 bool in[maxn];
 struct node
 {
     int l,r;
     int best,sum;
     int lc,rc;
 }seg[*maxn];

 inline int big(int a,int b){if (a > b)return a;return b;}
 inline void add(int,int);
 inline void dfs(int,int);
 inline void link(int,int);
 inline void build(int,int);
 inline void change(int,int);
 inline void find_best(int,int);
 inline void find_sum(int,int);
 inline int seg_sum(int,int,int);
 inline int seg_best(int,int,int);

 int main()
 {
     freopen("1036.in","r",stdin);
     freopen("1036.out","w",stdout);
     scanf("%d",&n);
     int i,a,b;
     for (i = ;i<n;i++)
     {
         scanf("%d %d",&a,&b);
         add(a,b); add(b,a);
     }
     for (i = ;i<=n;i++)
         scanf("%d",key+i);
     tot = ;
     dfs(root,);
     tot = ;
     memset(in,,sizeof(in));
     link(root,root);
     tot = ;
     build(,n);
     int T;
     scanf("%d\n",&T);
     char sign[];
     while (T)
     {
         T--;
         scanf("%s %d %d\n",sign,&a,&b);
         if (sign[] == 'Q')
         {
             if (sign[] == 'M')
                 find_best(a,b);
             if (sign[] == 'S')
                 find_sum(a,b);
         }
         else
             key[a] = b,change(a,);

     }
     return ;
 }

 inline void add(int a,int b)
 {
     tot++;
     toit[tot] = b;
     next[tot] = side[a];
     side[a] = tot;
 }

 inline void dfs(int a,int deep)
 {
     in[a] = true; dep[a] = deep;
     int u = side[a],v;
     son[a] = ;
     while (u != )
     {
         v = toit[u];
         if (!in[v])
         {
             father[v] = a;
             edge[++tot] = v;
             dfs(v,deep+);
             if (son[heavy[a]] < son[v])
                 heavy[a] = v;
             son[a] += son[v];
         }
         u = next[u];
     }
 }

 inline void link(int a,int high)
 {
     top[a] = high;
     pos[a] = ++tot;
     ord[tot] = a;
     in[a] = true;
     if (heavy[a] != )
         link(heavy[a],high);
     else return;
     int u = side[a],v;
     while (u != )
     {
         v = toit[u];
         if (!in[v])
             link(v,v);
         u = next[u];
     }
 }

 inline void build(int l,int r)
 {
     seg[tot].l = l;
     seg[tot].r = r;
     if (l == r)
     {
         seg[tot].best = key[ord[l]];
         seg[tot].sum = key[ord[l]];
         return;
     }
     int mid = ((l+r)>>);
     int k = tot;
     seg[k].lc = ++tot;
     build(l,mid);
     seg[k].rc = ++tot;
     build(mid+,r);
     seg[k].best = big(seg[seg[k].lc].best,seg[seg[k].rc].best);
     seg[k].sum = seg[seg[k].lc].sum+seg[seg[k].rc].sum;
 }

 inline void change(int a,int now)
 {
     if (seg[now].l == seg[now].r)
     {
         seg[now].best = key[a];
         seg[now].sum = key[a];
         return;
     }
     if (seg[seg[now].lc].l<=pos[a]&&seg[seg[now].lc].r>=pos[a])
         change(a,seg[now].lc);
     else change(a,seg[now].rc);
     seg[now].best = big(seg[seg[now].lc].best,seg[seg[now].rc].best);
     seg[now].sum = seg[seg[now].lc].sum + seg[seg[now].rc].sum;
 }

 inline void find_sum(int a,int b)
 {
     int sum = ;
     while (top[a] != top[b])
     {
         if (dep[top[a]]  >= dep[top[b]])
         {
             sum += seg_sum(pos[top[a]],pos[a],);
             a = father[top[a]];
         }
         else
         {
             sum += seg_sum(pos[top[b]],pos[b],);
             b = father[top[b]];
         }
     }
     if (pos[a] <= pos[b])
         sum += seg_sum(pos[a],pos[b],);
     else
         sum += seg_sum(pos[b],pos[a],);
     printf("%d\n",sum);
 }

 inline void find_best(int a,int b)
 {
     int best = -;
     while (top[a] != top[b])
     {
         if (dep[top[a]]  >= dep[top[b]])
         {
             best = big(best,seg_best(pos[top[a]],pos[a],));
             a = father[top[a]];
         }
         else
         {
             best = big(best,seg_best(pos[top[b]],pos[b],));
             b = father[top[b]];
         }
     }
     if (pos[a] <= pos[b])
         best = big(best,seg_best(pos[a],pos[b],));
     else
         best = big(best,seg_best(pos[b],pos[a],));
     printf("%d\n",best);
 }

 inline int seg_best(int l,int r,int now)
 {
     if (seg[now].l >=l && seg[now].r <=r)
         return seg[now].best;
     int ret =-;
     int t;
     t = seg[now].lc;
     if (seg[t].l<=l&&seg[t].r>=l)
         ret = big(ret,seg_best(l,r,t));
     else if (seg[t].l>=l&&seg[t].l<=r)
         ret = big(ret,seg_best(l,r,t));
     else if (seg[t].r>=l&&seg[t].r<=r)
         ret = big(ret,seg_best(l,r,t));
     t = seg[now].rc;
     if (seg[t].l<=l&&seg[t].r>=l)
         ret = big(ret,seg_best(l,r,t));
     else if (seg[t].l>=l&&seg[t].l<=r)
         ret = big(ret,seg_best(l,r,t));
     else if (seg[t].r>=l&&seg[t].r<=r)
         ret = big(ret,seg_best(l,r,t));
     return ret;
 }

 inline int seg_sum(int l,int r,int now)
 {
     if (seg[now].l >=l && seg[now].r <=r)
         return seg[now].sum;
     int ret = ;
     int t;
     t = seg[now].lc;
     if (seg[t].l<=l&&seg[t].r>=l)
         ret += seg_sum(l,r,t);
     else if (seg[t].l>=l&&seg[t].l<=r)
         ret += seg_sum(l,r,t);
     else if (seg[t].r>=l&&seg[t].r<=r)
         ret += seg_sum(l,r,t);
     t = seg[now].rc;
     if (seg[t].l<=l&&seg[t].r>=l)
         ret += seg_sum(l,r,t);
     else if (seg[t].l>=l&&seg[t].l<=r)
         ret += seg_sum(l,r,t);
     else if (seg[t].r>=l&&seg[t].r<=r)
         ret += seg_sum(l,r,t);
     return ret;
 }