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
解題思路:我們需要先算出來從0出發到每一個鏈上的節點的距離,然後我們從新標一下號,使用dfs序編號,那麼我必須經過x的話,就可以轉換爲我在區間(in[x],out[x])中取一個最大值,單點修改呢?
單點修改的話他只會對自己和自己下面的節點有貢獻,他下面有多少個節點,dfs序裏已經給出來了。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const long long inf=1e18;
const int maxn=1e5+10;
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long
int n,m,head[maxn],tot,in[maxn],out[maxn],cnt;
ll dis[maxn],b[maxn],sum[maxn<<2],lz[maxn<<2],cose[maxn];
struct Node{
int v,next;
}node[maxn<<2];
void init(){
mem(head,-1);tot=cnt=0;
}
void add(int u,int v){
node[tot].v=v;node[tot].next=head[u];head[u]=tot++;
}
void dfs(int u,int pre){
in[u]=++cnt;
b[cnt]=dis[u];
for(int i=head[u];~i;i=node[i].next){
int v=node[i].v;
if(v==pre) continue;
dis[v]=dis[u]+cose[v];
dfs(v,u);
}
out[u]=cnt;
}
void pushup(int k){
sum[k]=max(sum[k<<1],sum[k<<1|1]);
}
void pushdown(int k){
if(lz[k]){
lz[k<<1]+=lz[k];
lz[k<<1|1]+=lz[k];
sum[k<<1]+=lz[k];
sum[k<<1|1]+=lz[k];
lz[k]=0;
}
return ;
}
void build(int l,int r,int k){
lz[k]=0;
if(l==r){
sum[k]=b[l];
return ;
}
int mid=(l+r)>>1;
build(l,mid,k<<1);build(mid+1,r,k<<1|1);
pushup(k);
}
void add(int l,int r,int ql,int qr,ll val,int k){
if(l==ql&&r==qr){
sum[k]+=val;
lz[k]+=val;
return ;
}
pushdown(k);
int mid=(l+r)>>1;
if(qr<=mid) add(l,mid,ql,qr,val,k<<1);
else if(ql>mid) add(mid+1,r,ql,qr,val,k<<1|1);
else add(l,mid,ql,mid,val,k<<1),add(mid+1,r,mid+1,qr,val,k<<1|1);
pushup(k);
}
ll ans;
void query(int l,int r,int ql,int qr,int k){
if(ql==l&&qr==r){
ans=max(ans,sum[k]);
return ;
}
int mid=(l+r)>>1;
pushdown(k);
if(qr<=mid) query(l,mid,ql,qr,k<<1);
else if(ql>mid) query(mid+1,r,ql,qr,k<<1|1);
else query(l,mid,ql,mid,k<<1),query(mid+1,r,mid+1,qr,k<<1|1);
}
int main(){
int i,j,t,cas=1;
scanf("%d",&t);
while(t--){
init();
scanf("%d%d",&n,&m);
int u,ki;
ll v;
for(i=1;i<n;i++){
scanf("%d%d",&u,&v);
add(u,v);add(v,u);
}
for(i=0;i<n;i++)
scanf("%lld",&cose[i]);
dis[0]=cose[0];
dfs(0,-1);
build(1,n,1);
printf("Case #%d:\n",cas++);
while(m--){
scanf("%d",&ki);
if(ki==1){
scanf("%d",&u);
ans=-inf;
query(1,n,in[u],out[u],1);
printf("%lld\n",ans);
}
else{
scanf("%d%lld",&u,&v);
ll cha=v-cose[u];
add(1,n,in[u],out[u],cha,1);
cose[u]=v;
}
}
}
return 0;
}