二叉苹果树

在这里插入图片描述
我们可以把保留边转换成保留点(因为我个人对保留点熟)。
跟据树的性质,一个点有0or1个根。所以我们可以把边上的苹果数转移到子节点上。
根据题意可知这应是一道DP题(废话),设f[i][j]为以i为根的树上保留j个节点的最大权值和,又因为这是个二叉树,所以我们不妨设左子树要保留的节点数为k,那么右子树的需要保留的节点数为 j-1-k。
故状态转移方程为f[i][j]=max(f[i][j],f[li][k]+f[ri][j-1-k]+a[i])。

#include<iostream>
#include<cstdio>
using namespace std;
int n,q,t,a[150],l[150],f[150][150],r[150],head[110],to[210],next[210],w[210];
void df(int x,int fa)
{
	int i;
	for(i=head[x];i;i=next[i])
	{
		int v;
		v=to[i];
		if(v==fa) continue;
		if(!l[x]) l[x]=v;else r[x]=v;
		a[v]=w[i];
		df(v,x);
	}
}
int dp(int i,int j)
{
	if(j==0) return 0;
	if(l[i]==0) return a[i];
	if(f[i][j]>0) return f[i][j];
	for(int k=0;k<j;k++)
		f[i][j]=max(dp(l[i],k)+dp(r[i],j-k-1)+a[i],f[i][j]);
	return f[i][j];
}
int main()
{
	scanf("%d%d",&n,&q);q++;
	for(int i=1;i<n;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		t++;next[t]=head[x];head[x]=t;to[t]=y;w[t]=z;
		t++;next[t]=head[y];head[y]=t;to[t]=x;w[t]=z;
	}
	df(1,0);
	cout<<dp(1,q);
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值