(线段树+dfs序)

百度科技园内有nn个零食机,零食机之间通过n−1n−1条路相互连通。每个零食机都有一个值vv,表示为小度熊提供零食的价值。

由于零食被频繁的消耗和补充,零食机的价值vv会时常发生变化。小度熊只能从编号为0的零食机出发,并且每个零食机至多经过一次。另外,小度熊会对某个零食机的零食有所偏爱,要求路线上必须有那个零食机。

为小度熊规划一个路线,使得路线上的价值总和最大。
Input
输入数据第一行是一个整数T(T≤10)T(T≤10),表示有TT组测试数据。

对于每组数据,包含两个整数n,m(1≤n,m≤100000)n,m(1≤n,m≤100000),表示有nn个零食机,mm次操作。

接下来n−1n−1行,每行两个整数xx和y(0≤x,y小于n)y(0≤x,y小于n),表示编号为xx的零食机与编号为yy的零食机相连。

接下来一行由nn个数组成,表示从编号为0到编号为n−1n−1的零食机的初始价值v(|v|<100000)v(|v|<100000)。

接下来mm行,有两种操作:0 x y0 x y,表示编号为xx的零食机的价值变为yy;1 x1 x,表示询问从编号为0的零食机出发,必须经过编号为xx零食机的路线中,价值总和的最大值。

本题可能栈溢出,辛苦同学们提交语言选择c++,并在代码的第一行加上:

#pragma comment(linker, "/STACK:1024000000,1024000000")
Output
对于每组数据,首先输出一行”Case #?:”,在问号处应填入当前数据的组数,组数从1开始计算。

对于每次询问,输出从编号为0的零食机出发,必须经过编号为xx零食机的路线中,价值总和的最大值。
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

  • DFS序:把一棵树的前序遍历转化为区间来进行操作,分别存储一个节点子树的左端点和右端点(好像叫做时间戳)然后就能用线段树来进行查询了,在本题中,线段树存储的是从根节点出发,到现在节点经过的总价值,必须要经过某点,就是要求以该节点为根的子树节点中到0的最大值
    dfs序
#pragma comment(linker, "/STACK:1024000000,1024000000") `
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn = 100010;
ll segtree[maxn*20];
ll add[maxn*20];
ll b[maxn];
ll dis[maxn];
int cnt=0,pa=0;
int head[maxn];
int n,m;
void init()
{
    cnt=0;
    pa=0;
    memset(head,-1,sizeof(head));
}
struct Edge
{
    int to;
    int next;
}e[2*maxn];
void adde(int a,int b)
{
    e[pa].next=head[a];
    e[pa].to=b;
    head[a]=pa++;
}
struct Node
{
    int l,r;
    ll v;
}node[maxn*2];
void  dfs(int x,int fx)
{
    cnt++;
    b[cnt]=dis[x];
    node[x].l=cnt;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int cur=e[i].to;
        if(cur==fx)continue;
        dis[cur]=dis[x]+node[cur].v;
        dfs(cur,x);
    }
    node[x].r=cnt;
}
void PushDown(ll rt,ll l)
{
    if(add[rt]!=0)
    {
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        segtree[rt<<1]+=add[rt];
        segtree[rt<<1|1]+=add[rt];
        add[rt]=0;
    }
}

void plant(ll left,ll right,ll rt)
{
    add[rt]=0;
    if(left==right)
    {
        segtree[rt]=b[left];
        return;
    }
    ll mid=(left+right)>>1;
    plant(left,mid,rt<<1);
    plant(mid+1,right,rt<<1|1);
    segtree[rt]=max(segtree[rt<<1],segtree[rt<<1|1]);
}
ll ans;
void query(ll left,ll right,ll rt,ll L,ll R)
{
    if(L<=left&&right<=R)
    {
        if(ans<segtree[rt])ans=segtree[rt];
        return;
    }
    PushDown(rt,right-left+1);
    ll mid=(left+right)/2;
    if(R<=mid)
        query(left,mid,rt<<1,L,R);
    else if(L>mid)
        query(mid+1,right,rt<<1|1,L,R);
    else
    {
        query(left,mid,rt<<1,L,R);
        query(mid+1,right,rt<<1|1,L,R);
    }
}


void update(ll left,ll right,ll rt,ll l,ll r,ll up)
{
    if(l<=left&&r>=right)
    {
        segtree[rt]+=up;
        add[rt]+=up;
        return;
    }
    PushDown(rt,right-left+1);
    int mid=(left+right)>>1;
    if(l<=mid)update(left,mid,rt<<1,l,r,up);
    if(r>mid)update(mid+1,right,rt<<1|1,l,r,up);
    segtree[rt]=max(segtree[rt<<1],segtree[rt<<1|1]);
}
int main()
{
    int t;
    scanf("%d",&t);
    ll sans,frisk,chara;
    int ca=1;
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=0;i<n-1;i++)
        {
            scanf("%I64d%I64d",&sans,&frisk);
            sans++;
            frisk++;
            adde(sans,frisk);
            adde(frisk,sans);
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%I64d",&node[i].v);
        }
        node[1].l=1;
        dis[1]=node[1].v;
        dfs(1,-1);
        plant(1,n,1);
        printf("Case #%d:\n",ca++);
        for(int i=0;i<m;i++)
        {
            scanf("%I64d",&chara);
            if(chara==1)
            {
                scanf("%I64d",&frisk);
                frisk++;
                ans=-1e12;
                query(1,n,1,node[frisk].l,node[frisk].r);
                printf("%I64d\n",ans);

            }
            else
            {
                scanf("%I64d%I64d",&frisk,&sans);
                frisk++;
                update(1,n,1,node[frisk].l,node[frisk].r,sans-node[frisk].v);
                node[frisk].v=sans;
            }
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值