hdu 5692 (dfs序+线段树）

Snacks

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

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

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

 

Source

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

 

Recommend

wange2014   |   We have carefully selected several similar problems for you:  5932 5931 5930 5929 5928


 题解：万万没想到会爆int，导致查了好久的错，
先用dfs处理出一个从0到每个点的距离，那么经过一个点的最大值就是它的最大的某个子节点到0的距离，于是利用dfs序把题目改成线段树的区间更新和查询问题
代码

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 100005
long long l[MAXN],r[MAXN],pre[MAXN],a[MAXN],add[MAXN*4],info[MAXN],nd[MAXN];
vector<long long> to,nxt;
struct node
{
    long long v,left,right;
}tree[4*MAXN];
void push_down(long long x)
{
    if(add[x])
    {
        tree[x*2].v+=add[x];
        tree[x*2+1].v+=add[x];
        add[x*2]+=add[x];
        add[x*2+1]+=add[x];
        add[x]=0;
    }
}
void build(long long id,long long l,long long r)
{
    tree[id].left=l,tree[id].right=r;
    if(l==r)
    {
        tree[id].v=pre[nd[l]];
        return;
    }
    long long mid=(tree[id].left+tree[id].right)/2;
    build(id*2,l,mid);
    build(id*2+1,mid+1,r);
    tree[id].v=max(tree[id*2].v,tree[id*2+1].v);
}
void update(long long id,long long l,long long r,long long x)
{
    if(tree[id].left==l&&tree[id].right==r)
    {
        tree[id].v+=x;
        add[id]+=x;
        return;
    }
    push_down(id);
    long long mid=(tree[id].left+tree[id].right)/2;
    if(mid<l)
        update(id*2+1,l,r,x);
    else if(mid>=r)
        update(id*2,l,r,x);
    else
    {
        update(id*2,l,mid,x);
        update(id*2+1,mid+1,r,x);
    }
    tree[id].v=max(tree[id*2].v,tree[id*2+1].v);
}
long long query(long long id,long long L,long long R)
{
    if(tree[id].left==L&&tree[id].right==R)
    {
        return tree[id].v;
    }
    push_down(id);
    long long mid=(tree[id].left+tree[id].right)/2;
    if(mid<L)
        return query(id*2+1,L,R);
    else if(mid>=R)
        return query(id*2,L,R);
    else
        return max(query(id*2,L,mid),query(id*2+1,mid+1,R));
}
void Add(long long a,long long b)
{
    to.push_back(b);
    nxt.push_back(info[a]);
    info[a]=to.size()-1;
}
long long cnt;
void dfs(long long x,long long fa)
{
    cnt++;
      l[x]=cnt;
       nd[cnt]=x;
    for(long long i=info[x];i!=-1;i=nxt[i])
    {
        if(to[i]!=fa)
        {
            pre[to[i]]=pre[x]+a[to[i]];
            dfs(to[i],x);
        }
    }
    r[x]=cnt;
}
void init()
{
    cnt=0;

    memset(info,-1,sizeof(info));
    memset(add,0,sizeof(add));
    to.clear();
    nxt.clear();
}
int main()
{
    long long t;
    long long cas=1;
    scanf("%I64d",&t);
    while(cas<=t)
    {
        init();
        long long n,m;
        scanf("%I64d%I64d",&n,&m);
        for(long long i=0;i<n-1;i++)
        {
            long long a,b;
            scanf("%I64d%I64d",&a,&b);
            Add(a,b);
            Add(b,a);
        }
        for(long long i=0;i<n;i++)
        scanf("%I64d",&a[i]);
        pre[0]=a[0];
        dfs(0,-1);
        build(1,1,n);
        printf("Case #%I64d:\n",cas++);
        for(long long i=0;i<m;i++)
        {
            long long op;
            scanf("%I64d",&op);
            if(op==0)
            {
                long long x,y;
                scanf("%I64d%I64d",&x,&y);
                update(1,l[x],r[x],y-a[x]);
                a[x]=y;
            }
            else
            {
                long long x;
                scanf("%I64d",&x);
                long long ans=query(1,l[x],r[x]);
                printf("%I64d\n",ans);
            }
        }
    }
    return 0;
}