HDU5692 线段树+DFS序+前缀和

题目链接

一、题目描述

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

二、解题思路

本题主要的算法为线段树、DFS序、前缀和。我们先采用DFS序，将整个树进行规整化，那么对于每个结点，其进入的时间戳与退出的时间戳，可以用来查询这个区间。在线段树中，我们维护了一个前缀和，表示到这个点位置这条链上的前缀和。显然取前缀和最大的即可。

因此采用线段树维护最大的前缀和，区间修改。

线段树模板改自此

三、Code

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define N 100010
#define LD (o << 1)
#define RD (o << 1 | 1)
#define INF 1e18
struct Node
{
    int l, r;
    ll maxv, add;
};
class Tree
{
    public:Node t[N << 2];
    public:ll a[N];
    inline void pushup(Node &o, Node &ld, Node &rd)
    {
        o.maxv = max(ld.maxv, rd.maxv);
    }
    inline void pushdown(Node &o, Node &ld, Node &rd)
    {
        ld.add += o.add;
        ld.maxv += o.add;
        rd.add += o.add;
        rd.maxv += o.add;
        o.add = 0;
    }
    void build(int o, int l, int r)
    {
        t[o].l = l;
        t[o].r = r;
        t[o].add = 0;
        if (l == r)
        {
            t[o].maxv = a[l];
            return;
        }
        int mid = (l + r) >> 1;
        build(LD, l, mid);
        build(RD, mid + 1, r);
        pushup(t[o], t[LD], t[RD]);
    }
    void update(int o, int l, int r, ll v)
    {
        if (l <= t[o].l && t[o].r <= r)
        {
            t[o].add += v;
            t[o].maxv += v;
            return;
        }
        pushdown(t[o], t[LD], t[RD]);
        int mid = (t[o].l + t[o].r) >> 1;
        if (mid >= l)
            update(LD, l, r, v);
        if (mid < r)
            update(RD, l, r, v);
        pushup(t[o], t[LD], t[RD]);
    }
    ll query(int o, int l, int r)
    {
        if (l <= t[o].l && t[o].r <= r)
            return t[o].maxv;
        pushdown(t[o], t[LD], t[RD]);
        ll ans = -INF;
        int mid = (t[o].l + t[o].r) >> 1;
        if (mid >= l)
            ans = max(ans, query(LD, l, r));
        if (mid < r)
            ans = max(ans, query(RD, l, r));
        pushup(t[o], t[LD], t[RD]);
        return ans;
    }
}A;
int n, m;
struct point{
    int in,out;
}p[100005];
vector<vector<int> > vec(100005);
ll orgnum[100005];
ll pre[100005];
int tim=0;
ll ss=0;
void dfs(int now,int fa){
    tim++;
    p[now].in=tim;
    ss+=orgnum[now];
    A.a[tim]=ss;
    for(int i=0;i<vec[now].size();i++){
        int to=vec[now][i];
        if(to==fa)
            continue;
        dfs(to,now);
    }
    ss-=orgnum[now];
    p[now].out=tim;
}
int main()
{
    int T;
    scanf("%d",&T);
    for (int tc = 1; tc <= T; tc++)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;i++){
            pre[i]=0;
            vec[i].clear();
        }
        for (int i = 1; i < n; i++)
        {
            int aa, bb;
            scanf("%d%d", &aa, &bb);
            vec[aa].push_back(bb);
            vec[bb].push_back(aa);
        }
        for(int i=0;i<n;i++){
            scanf("%lld",&orgnum[i]);
        }
		A = Tree();
        tim=0;
        ss=0;
        dfs(0,0);
        
        A.build(1, 1, n);
        printf("Case #%d:\n",tc);
        for (int i = 1; i <= m; i++)
        {
            int x,y;ll z;
            scanf("%d",&x);
            if (x == 1)
            {
                scanf("%d",&y);
                printf("%lld\n",A.query(1,p[y].in,p[y].out));
            }
            if (x == 0){
                scanf("%d %lld",&y,&z);
                ll d=z-orgnum[y];
                orgnum[y]=z;
                A.update(1,p[y].in,p[y].out,d);
            }
        }
    }
}