HDU - 5692 Snacks(dfs序+线段树)

Snacks

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

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

 思路:dfs序我们求出到达没个点的前缀和，然后我只需要查询in[x]到out[x]最大值及经过x点的最大值。在我们进行更新操作时，我们将某点值改成y时，就是加上y-cost[x]。

然后区间维护下最大值就行了~  发现树状结构+线段树解问题 似乎很强大emmm

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define pb push_back
#define P pair<int,int>
#define INF 1e18
using namespace std;
const int maxn = 100010;
int head[maxn],Next[maxn*200],To[maxn*200];
int in[maxn],out[maxn],cnt,tot;
ll sum[maxn<<2],lazy[maxn<<2],cost[maxn],dis[maxn],pos[maxn];
void init()
{
    memset(head,-1,sizeof(head));
    memset(dis,0,sizeof(dis));
    cnt=tot=0;
}
void add(int u,int v)
{
    Next[++cnt]=head[u];
    head[u]=cnt;
    To[cnt]=v;
}
void dfs(int u,int fa)
{
    in[u]=++tot;
    pos[tot]=dis[u];
    for(int i=head[u]; i!=-1; i=Next[i])
    {
        if(To[i]==fa) continue;
        dis[To[i]]=dis[u]+cost[To[i]];
        dfs(To[i],u);
    }
    out[u]=tot;
}
void push_up(int rt)
{
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
void push_down(int rt)
{
    if(lazy[rt])
    {
        lazy[rt<<1]+=lazy[rt];
        lazy[rt<<1|1]+=lazy[rt];
        sum[rt<<1]+=lazy[rt];
        sum[rt<<1|1]+=lazy[rt];
        lazy[rt]=0;
    }
}
void build(int l,int r,int rt)
{
    lazy[rt]=0;
    if(l==r)
    {
        sum[rt]=pos[l];
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    push_up(rt);
}
void updata(int l,int r,int rt,int L,int R,ll val)
{
    if(L<=l&&r<=R)
    {
        sum[rt]+=val;
        lazy[rt]+=val;
        return;
    }
    push_down(rt);
    int m=(l+r)>>1;
    if(L<=m) updata(lson,L,R,val);
    if(R>m) updata(rson,L,R,val);
    push_up(rt);
}
ll query(int l,int r,int rt,int L,int R)
{
    if(L<=l&&r<=R)
        return sum[rt];
    push_down(rt);
    int m=(l+r)>>1;
    ll ans=-INF;
    if(L<=m) ans=max(ans,query(lson,L,R));
    if(R>m) ans=max(ans,query(rson,L,R));
    return ans;
}
int main()
{
    int t,n,m,u,v,k=0;
    scanf("%d",&t);
    while(t--)
    {
        printf("Case #%d:\n",++k);
        init();
        scanf("%d %d",&n,&m);
        for(int i=1; i<n; i++)
        {
            scanf("%d %d",&u,&v);
            add(u,v);
            add(v,u);
        }
        for(int i=0; i<n; i++)
            scanf("%lld",&cost[i]);
        dis[0]=cost[0];
        dfs(0,-1);
        build(1,n,1);
        while(m--)
        {
            int op,x,y;
            scanf("%d %d",&op,&x);
            if(op==0)
            {
                scanf("%d",&y);
                ll ans=1LL*y-cost[x];
                updata(1,n,1,in[x],out[x],ans);
                cost[x]=y;
            }
            else
            {
                printf("%lld\n",query(1,n,1,in[x],out[x]));
            }
        }
    }
}

View Code

 

 

 

PS:摸鱼怪的博客分享，欢迎感谢各路大牛的指点~

转载于:https://www.cnblogs.com/MengX/p/9498605.html