HDU5692：Snacks（DFS序 & 线段树区间更新）

Snacks

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 2796    Accepted Submission(s): 646

Problem 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

16 50 11 20 33 45 37 -5 100 20 -5 -71 11 30 2 -11 11 5

 

Sample Output

Case #1:10227220

 

Source

2016"百度之星" - 初赛（Astar Round2A）

思路：DFS序给节点重编号，每次查询从目标节点的子树用线段树取得最大值即可。

# include <iostream># include <cstdio># include <cstring># define lson l,m,id<<1# define rson m+1,r,id<<1|1using namespace std;typedef long long LL;const int maxn = 1e5+8;const LL INF = 0x3f3f3f3f3f3f3f3f;struct node{    int v, next;}edge[maxn<<1];LL ans[maxn<<2], lazy[maxn<<2], val[maxn], dis[maxn];int in[maxn], out[maxn], Next[maxn], New[maxn], cnt, cnt2;void add_edge(int u, int v){    edge[cnt] = {v,Next[u]};    Next[u] = cnt++;    edge[cnt] = {u,Next[v]};    Next[v] = cnt++;}void dfs(int u, int pre, LL d){    dis[u] = d;    in[u] = ++cnt2;    New[cnt2] = u;    for(int i=Next[u]; i!=-1; i=edge[i].next)    {        int v = edge[i].v;        if(v==pre) continue;        dfs(v, u, d+val[v]);    }    out[u] = cnt2;}void build(int l, int r, int id){    lazy[id] = 0;    if(l==r)    {        ans[id] = dis[New[l]];        return;    }    int m = l+r>>1;    build(lson);    build(rson);    ans[id] = max(ans[id<<1], ans[id<<1|1]);}void pushdown(int id){    if(lazy[id])    {        lazy[id<<1] += lazy[id];        lazy[id<<1|1] += lazy[id];        ans[id<<1] += lazy[id];        ans[id<<1|1] += lazy[id];        lazy[id] = 0;    }}LL query(int L, int R, int l, int r, int id){    if(L<=l&&R>=r)        return ans[id];    pushdown(id);    int m = l+r>>1;    LL imax = -INF;    if(L<=m) imax = max(imax, query(L ,R, lson));    if(R>m) imax = max(imax, query(L, R, rson));    return imax;}void update(LL v, int L, int R, int l ,int r, int id){    if(L<=l&&R>=r)    {        ans[id] += v;        lazy[id] += v;        return;    }    pushdown(id);    int m = l+r>>1;    if(L<=m) update(v, L, R, lson);    if(R>m) update(v, L, R, rson);    ans[id] = max(ans[id<<1], ans[id<<1|1]);}int main(){    int n, m, t, a, b, cas=1;    scanf("%d",&t);    while(t--)    {        printf("Case #%d:\n",cas++);        cnt = cnt2 = 0;        memset(Next, -1, sizeof(Next));        scanf("%d%d",&n,&m);        for(int i=1; i<n; ++i)        {            scanf("%d%d",&a,&b);            add_edge(a+1, b+1);        }        for(int i=1; i<=n; ++i) scanf("%lld",&val[i]);        dfs(1, 0, val[1]);        build(1, n, 1);        while(m--)        {            int op;            scanf("%d",&op);            if(op == 0)            {                scanf("%d%d",&a,&b); ++a;                update(b-val[a], in[a], out[a], 1, n, 1);                val[a] = b;            }            else            {                scanf("%d",&a); ++a;                printf("%lld\n",query(in[a], out[a], 1, n, 1));            }        }    }    return 0;}