Bstar2016的一道水题T3

最新推荐文章于 2020-06-12 22:49:11 发布

bblss123

最新推荐文章于 2020-06-12 22:49:11 发布

阅读量681

点赞数 1

分类专栏： Bestcoder 图论 |-----dfs序数据结构 |-----线段树文章标签： Bestcoder

本文链接：https://blog.youkuaiyun.com/bblss123/article/details/51472409

版权

图论同时被 3 个专栏收录

18 篇文章

订阅专栏

数据结构

13 篇文章

订阅专栏

Bestcoder

5 篇文章

订阅专栏

本文介绍了一种基于深度优先搜索和线段树的数据结构来解决百度科技园内的零食机路径规划问题，目标是找到一条从起点到指定零食机的路径，使得路径上零食机的价值总和最大。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Snacks

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

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

为小度熊规划一个路线，使得路线上的价值总和最大。

Solution
一道水题还想了蛮久==
首先dfs处理出dfs序以及每个点到根结点的距离。
然后建一棵记录距离的线段树。
然后就好了
记得会爆int

[源代码]

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#pragma comment(linker, "/STACK:1024000000,1024000000") 
using namespace std;
typedef long long ll;
const int M=1e5+5;
int A[M],tid[M],nxt[M],dfs_clock,tst,_;
ll d[M];
vector<int>G[M];
struct Segment_Tree{
    struct node{
        int l,r;
        ll mx,add;
    }tree[M<<2];
    inline void up(int p){
        tree[p].mx=max(tree[p<<1].mx,tree[p<<1|1].mx);
    }
    inline void down(int p){
        if(!tree[p].add)return;
        tree[p<<1].add+=tree[p].add;tree[p<<1|1].add+=tree[p].add;
        tree[p<<1].mx+=tree[p].add,tree[p<<1|1].mx+=tree[p].add;
        tree[p].add=0;
    }
    void build(int l,int r,int p){
        tree[p].l=l,tree[p].r=r,tree[p].mx=tree[p].add=0;
        if(l==r){
            tree[p].mx=d[l];
            return;
        }
        int mid=l+r>>1;
        build(l,mid,p<<1);
        build(mid+1,r,p<<1|1);
        up(p);
    }
    void update(int l,int r,int x,int p){
        if(tree[p].l==l&&tree[p].r==r){
            tree[p].add+=x,tree[p].mx+=x;
            return;
        }
        down(p);
        int mid=tree[p].l+tree[p].r>>1;
        if(l>mid)update(l,r,x,p<<1|1);
        else if(r<=mid)update(l,r,x,p<<1);
        else update(l,mid,x,p<<1),update(mid+1,r,x,p<<1|1);
        up(p);
    }
    ll query(int l,int r,int p){
        if(tree[p].l==l&&tree[p].r==r)return tree[p].mx;
        down(p);
        int mid=tree[p].l+tree[p].r>>1;
        if(l>mid)return query(l,r,p<<1|1);
        if(r<=mid)return query(l,r,p<<1);
        return max(query(l,mid,p<<1),query(mid+1,r,p<<1|1));
    }
}T;
void dfs(int v,int f,ll dep){
    tid[v]=++dfs_clock,d[tid[v]]=dep;
    for(int i=0;i<G[v].size();++i)
        if(G[v][i]^f)dfs(G[v][i],v,dep+A[G[v][i]]);
    nxt[v]=dfs_clock;
}
void gao(){
    int n,q;cin>>n>>q;
    dfs_clock=0;
    memset(d,0,sizeof(d));
    for(int i=0;i^n;++i)G[i].clear();
    for(int i=1,a,b;i^n;++i)scanf("%d %d",&a,&b),G[a].push_back(b),G[b].push_back(a);
    for(int i=0;i^n;++i)scanf("%d",&A[i]);
    printf("Case #%d:\n",++tst);
    dfs(0,0,A[0]);
    T.build(1,n,1);
    for(int opt,x,k;q--;){
        scanf("%d %d",&opt,&x);
        switch(opt){
            case(0):scanf("%d",&k);T.update(tid[x],nxt[x],k-A[x],1);A[x]=k;break;
            case(1):printf("%I64d\n",T.query(tid[x],nxt[x],1));break;
        }
    }
}
int main(){for(cin>>_;_--;)gao();}