HDU 5692 Snacks dfs序 + 线段树

题目：http://acm.hdu.edu.cn/showproblem.php?pid=5692

题意：

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零食机的路线中，价值总和的最大值。

思路：2016百度之星初赛题目，当时以为是树链剖分 + 线段树，但是不会做-_-||。。。原来是dfs序 + 线段树。首先把树形转换为线性，我们用dfs序给每个点编号，in[v] 到 out[v]之间的编号包括了v的所有子孙，这道题就是求0到v的子孙的最大距离，于是变成了查询区间最大值，更新时区间更新。

总结：这题long long数据是个坑。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;

typedef long long ll;
const int N = 100010;
const ll INF = 1e18;
struct edge
{
    int to, next;
}g[N*2];
struct node
{
    int l, r;
    ll val, mark;
}s[N*4];
int head[N], in[N], out[N];
ll tmp[N], val[N];
int n, m, cnt, num, _case = 0;
void add_edge(int v, int u)
{
    g[cnt].to = u;
    g[cnt].next = head[v];
    head[v] = cnt++;
}
void dfs(int v, int fa, ll w)
{
    in[v] = ++num, val[num] = w;
    for(int i = head[v]; i != -1; i = g[i].next)
    {
        int u = g[i].to;
        if(u != fa)
            dfs(u, v, w + tmp[u]);
    }
    out[v] = num;
}
void push_up(int k)
{
    s[k].val = max(s[k<<1].val, s[k<<1|1].val);
}
void push_down(int k)
{
    if(s[k].mark)
    {
        s[k<<1].val += s[k].mark;
        s[k<<1|1].val += s[k].mark;
        s[k<<1].mark += s[k].mark;
        s[k<<1|1].mark += s[k].mark;
        s[k].mark = 0;
    }
}
void build(int l, int r, int k)
{
    s[k].l = l, s[k].r = r, s[k].mark = 0;
    if(l == r)
    {
        s[k].val = val[l];
        return;
    }
    int mid = (l + r) >> 1;
    build(l, mid, k << 1);
    build(mid + 1, r, k << 1|1);
    push_up(k);
}
void update(int l, int r, ll c, int k)
{
    if(l <= s[k].l && s[k].r <= r)
    {
        s[k].val += c;
        s[k].mark += c;
        return;
    }
    push_down(k);
    int mid = (s[k].l + s[k].r) >> 1;
    if(l <= mid) update(l, r, c, k << 1);
    if(r > mid) update(l, r, c, k << 1|1);
    push_up(k);
}
ll query(int l, int r, int k)
{
    if(l <= s[k].l && s[k].r <= r)
        return s[k].val;
    push_down(k);
    int mid = (s[k].l + s[k].r) >> 1;
    ll ans = -INF;
    if(l <= mid) ans = max(ans, query(l, r, k << 1));
    if(r > mid) ans = max(ans, query(l, r, k << 1|1));
    return ans;
}
int main()
{
    int t, a, b;
    ll c;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &m);
        cnt = num = 0;
        memset(head, -1, sizeof head);
        for(int i = 0; i < n - 1; i++)
        {
            scanf("%d%d", &a, &b);
            add_edge(a, b);
            add_edge(b, a);
        }
        for(int i = 0; i < n; i++)
            scanf("%I64d", &tmp[i]);
        dfs(0, -1, tmp[0]);
        build(1, num, 1);
        printf("Case #%d:\n", ++_case);
        for(int i = 0; i < m; i++)
        {
            scanf("%d%d", &a, &b);
            if(a == 1) printf("%I64d\n", query(in[b], out[b], 1));
            else
            {
                scanf("%I64d", &c);
                update(in[b], out[b], c - tmp[b], 1);
                tmp[b] = c;
            }
        }
    }
    return 0;
}