Bstar2016的一道水题T3

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();}

竟然在C++里跑第一==

一定是交的人太少了==