HDU 5692 Snacks

Snacks

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

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:  6216 6215 6214 6213 6212 

 

思路：dfs序+线段树。用线段树维护从根节点到每一个节点的距离：修改操作就是x子树对应区间加上一个值，查询操作就是查询x子树对应区间的最大值。

错因：

第一遍MLE：数组开小了。

第二遍RE爆栈：题目有提示手动开栈没有开。

第三遍RE数组越界：线段树数组开小了。

第四遍WA：数组没有用long long。

第五遍WA：在修改后val[x]忘记重新赋值为y

第六遍WA：输入输出没有改用"%lld"。

第七遍WA：查询的函数设为了int，返回值应为long long。

第八遍：终于AC啦~(≧▽≦)/~啦啦啦哈哈哈哈哈哈哈哈哈

第九遍：发现可以不用手动开栈AC

//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 100010
using namespace std;
struct nond{
    int l,r;
    long long dis,falg;
}tree[MAXN*8];
int T,n,m,sz,tot,num;
long long val[MAXN*2];
int idl[MAXN*2],idr[MAXN*2],dad[MAXN*2];
int to[MAXN*2],net[MAXN*2],head[MAXN*2];
void add(int u,int v){
    to[++tot]=v;net[tot]=head[u];head[u]=tot;
    to[++tot]=u;net[tot]=head[v];head[v]=tot;
}
void up(int now){
    tree[now].dis=max(tree[now*2].dis,tree[now*2+1].dis);
}
void build(int now,int l,int r){
    tree[now].l=l;tree[now].r=r;
    if(tree[now].l==tree[now].r)
        return ;
    int mid=(tree[now].l+tree[now].r)/2;
    build(now*2,l,mid);
    build(now*2+1,mid+1,r);
}
void down(int now){
    tree[now*2].falg+=tree[now].falg;
    tree[now*2+1].falg+=tree[now].falg;
    tree[now*2].dis+=tree[now].falg;
    tree[now*2+1].dis+=tree[now].falg; 
    tree[now].falg=0;
}
void change(int now,int l,int r,long long k){
    if(tree[now].l==l&&tree[now].r==r){
        tree[now].dis+=k;
        tree[now].falg+=k;
        return ;
    }
    if(tree[now].falg)    down(now);
    int mid=(tree[now].l+tree[now].r)/2;
    if(r<=mid)    change(now*2,l,r,k);
    else if(l>mid)    change(now*2+1,l,r,k);
    else{
        change(now*2,l,mid,k);
        change(now*2+1,mid+1,r,k);
    }
    up(now);
}
long long query(int now,int l,int r){
    if(tree[now].l==l&&tree[now].r==r)
        return tree[now].dis;
    if(tree[now].falg)    down(now);
    int mid=(tree[now].l+tree[now].r)/2;
    if(r<=mid)    return query(now*2,l,r);
    else if(l>mid)    return query(now*2+1,l,r);
    else    return max(query(now*2,l,mid),query(now*2+1,mid+1,r));
}
void dfs(int now){
    idl[now]=++sz;
    for(int i=head[now];i;i=net[i])
        if(to[i]!=dad[now]){
            dad[to[i]]=now;
            dfs(to[i]);
        }
    idr[now]=++sz;
}
int main(){
    scanf("%d",&T);
    while(T--){
        sz=0;tot=0;num++;
        memset(val,0,sizeof(val));
        memset(dad,0,sizeof(dad));
        memset(idl,0,sizeof(idl));
        memset(idr,0,sizeof(idr));
        memset(head,0,sizeof(head));
        memset(tree,0,sizeof(tree));
        cout<<"Case #"<<num<<":"<<endl;
        scanf("%d%d",&n,&m);
        for(int i=1;i<n;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            add(x,y);
        }
        dfs(0);
        build(1,1,sz);
        for(int i=0;i<n;i++){
            scanf("%lld",&val[i]);
            change(1,idl[i],idr[i],val[i]);
        }
        for(int i=1;i<=m;i++){
            int opt,x;long long y;
            scanf("%d%d",&opt,&x);
            if(opt==0){
                scanf("%lld",&y);
                change(1,idl[x],idr[x],y-val[x]);
                val[x]=y;
            }
            else if(opt==1)
                cout<<query(1,idl[x],idr[x])<<endl;
        }
    }
}