HDU 5692 Snacks

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章