HDU 5692 Snacks

版权声明:转载请说明出处 https://blog.csdn.net/bao___zi/article/details/82011837

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

猜你喜欢

转载自blog.csdn.net/bao___zi/article/details/82011837