poj 2486 树形dp(吃苹果)

题意:给定一颗有n个顶点的树,每个点上有一定数量的苹果。求从1号节点出发,最多可以走m步,最多能吃到的苹果数量。

思路:树形dp。dp[i][j][0]表示以i节点为根,最多走j步且最终不必回到i节点这种情况下能够吃到的最多苹果;dp[i][j][1]表示以i节点为根,最多走j步且最终必须回到i节点这种情况下能够吃到的最多苹果。对于dp[i][j][0],又分为两种情况:当前遍历到的儿子不必回到i,之前的必须回到i以及反之。

#include <stdio.h>
#include <string.h>
#define max(a,b) ((a)>(b)?(a):(b))
#define N 105
struct edge{
	int y,next;
}e[N*2];
int first[N],dp[N][205][2],num[N];//0表示不回来,1表示回来
int n,m,top;
void add(int x,int y){
	e[top].y = y;
	e[top].next = first[x];
	first[x] = top++;
}
void dfs(int x,int fa){
	int i,k,y,j;
	dp[x][0][1] = dp[x][0][0] = num[x];
	for(i = first[x];i!=-1;i=e[i].next){
		y = e[i].y;
		if(y == fa)
			continue;
		dfs(y,x);
		for(j = m;j>=0;j--){
			for(k = 0;k+j<=m;k++){
				if(k>=2){
					dp[x][k+j][1]=max(dp[x][k+j][1],dp[x][j][1]+dp[y][k-2][1]);//必须回到i节点
					dp[x][k+j][0]=max(dp[x][k+j][0],dp[x][j][0]+dp[y][k-2][1]);//不必回到i节点,且当前的这支儿子必须回来
				}
				if(k>=1)
					dp[x][k+j][0]=max(dp[x][k+j][0],dp[x][j][1]+dp[y][k-1][0]);//不必回到i节点,且当前的这支儿子不用回来(也就是k-1的意思,因为只用去到儿子节点而不用它回来)
			}
		}
	}
}
int main(){
	freopen("a.txt","r",stdin);
	while(scanf("%d %d",&n,&m)!=EOF){
		int i,a,b;
		top = 0;
		memset(first,-1,sizeof(first));
		memset(dp,0,sizeof(dp));
		for(i = 1;i<=n;i++)
			scanf("%d",&num[i]);
		for(i = 1;i<n;i++){
			scanf("%d %d",&a,&b);
			add(a,b);add(b,a);
		}
		if(m>2*(n-1))
			m = 2*(n-1);
		dfs(1,0);
		printf("%d\n",dp[1][m][0]);
	}
	return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值