HDU5692(线段树+dfs序）

Snacks

Time Limit:5000MS     Memory Limit:65536KB     64bit IO Format:%I64d & %I64u

Submit Status

Description

百度科技园内有n个零食机，零食机之间通过n−1条路相互连通。每个零食机都有一个值v，表示为小度熊提供零食的价值。

由于零食被频繁的消耗和补充，零食机的价值v会时常发生变化。小度熊只能从编号为0的零食机出发，并且每个零食机至多经过一次。另外，小度熊会对某个零食机的零食有所偏爱，要求路线上必须有那个零食机。

为小度熊规划一个路线，使得路线上的价值总和最大。 

Input

输入数据第一行是一个整数T(T≤10)，表示有T组测试数据。

对于每组数据，包含两个整数n,m(1≤n,m≤100000)，表示有n个零食机，m次操作。

接下来n−1行，每行两个整数x和y(0≤x,y<n)，表示编号为x的零食机与编号为y的零食机相连。

接下来一行由n个数组成，表示从编号为0到编号为n−1的零食机的初始价值v(|v|<100000)。

接下来m行，有两种操作：0 x y，表示编号为x的零食机的价值变为y；1 x，表示询问从编号为0的零食机出发，必须经过编号为x零食机的路线中，价值总和的最大值。

本题可能栈溢出，辛苦同学们提交语言选择c++，并在代码的第一行加上：

`#pragma comment(linker, "/STACK:1024000000,1024000000") `

Output

对于每组数据，首先输出一行”Case #?:”，在问号处应填入当前数据的组数，组数从1开始计算。

对于每次询问，输出从编号为0的零食机出发，必须经过编号为x零食机的路线中，价值总和的最大值。 

Sample Input

1
6 5
0 1
1 2
0 3
3 4
5 3
7 -5 100 20 -5 -7
1 1
1 3
0 2 -1
1 1
1 5

Sample Output

Case #1:
102
27
2
20

线段树＋dfs序

 //2016.8.11
 #pragma comment(linker, "/STACK:1024000000, 1024000000")
 #include<iostream>
 #include<cstdio>
 #include<vector>
 #include<algorithm>
 #define N 100005
 #define lson (id<<1)
 #define rson ((id<<1)|1)
 #define mid ((l+r)>>1)

 using namespace std;

 struct node
 {
     long long max, lazy;
 }tree[N*];

 vector<int> v[N];
 long long w[N], val[N];//val[i]表示从根节点走到i节点的价值和
 int in[N], out[N], flag;//in记录查询区间的左端，out记录查询区间的右端

 //线段树模板
 //************************************************************************************
 void push_up(int id)
 {
     tree[id].max = max(tree[lson].max, tree[rson].max);
     return;
 }

 void build_tree(int id, int l, int r)
 {
     if(l == r)
     {
         tree[id].max = val[l];
         tree[id].lazy = ;
         return ;
     }
     build_tree(lson, l, mid);
     build_tree(rson, mid+, r);
     push_up(id);
     tree[id].lazy = ;
     return ;
 }

 void push_down(int id, int l, int r)
 {
     if(tree[id].lazy)
     {
         tree[lson].lazy += tree[id].lazy;
         tree[rson].lazy += tree[id].lazy;
         tree[lson].max += tree[id].lazy;
         tree[rson].max += tree[id].lazy;
         tree[id].lazy = ;
     }
     return ;
 }

 void ins(int id, int l, int r, int ql, int qr, int tt)
 {
     if(ql<=l&&r<=qr)
     {
         tree[id].max += tt;
         tree[id].lazy += tt;
         return;
     }
     if(tree[id].lazy) push_down(id, l, r);
     if(ql<=mid)
           ins(lson, l, mid, ql, qr, tt);
     if(mid+<=qr)
           ins(rson, mid+, r, ql, qr, tt);
     push_up(id);
     return;
 }

 long long query(int id, int l, int r, int ql, int qr)
 {
     if(ql<=l && r<=qr)
     {
         return tree[id].max;
     }
     if(tree[id].lazy)
           push_down(id, l, r);
     long long maxnum = -;
     if(ql<=mid) maxnum = max(maxnum, query(lson, l, mid, ql, qr));
     if(mid+<=qr) maxnum = max(maxnum, query(rson, mid+, r, ql, qr));
     return maxnum;
 }
 //************************************************************************************

 void dfs(int a, int fa, long long wight)
 {
     in[a] = ++flag;
     for(int i = ; i < v[a].size(); i++)
     {
         int b = v[a][i];
         if(b==fa)continue;
         dfs(b, a, w[b]+wight);
     }
     out[a] = flag;
     val[in[a]] = wight;
 }

 int main()
 {
     int T, n, m, a, b, cmd;
     cin>>T;
     for(int kase = ; kase <= T; kase++)
     {
         printf("Case #%d:\n", kase);
         cin>>n>>m;
         for(int i = ; i < N; i++)
             v[i].clear();
         for(int i = ; i < n-; i++)
         {
             scanf("%d%d", &a, &b);
             a++; b++;
             v[a].push_back(b);
             v[b].push_back(a);
         }
         for(int i =  ; i <= n; i++)
           scanf("%lld", &w[i]);
         flag = ;
         dfs(, , w[]);
         for(int i = ; i <= n; i++)
           cout<<in[i]<<" "<<out[i]<<endl;
         build_tree(, , n);
         while(m--)
         {
             scanf("%d", &cmd);
             if(cmd == )
             {
                 scanf("%d%d", &a, &b);a++;
                 ins(, , n, in[a], out[a], b-w[a]);
                 w[a] = b;
             }else
             {
                 scanf("%d", &a); a++;
                 long long ans = query(, , n, in[a], out[a]);
                 cout<<ans<<endl;
             }
         }
     }

     return ;
 }