我们可以把保留边转换成保留点(因为我个人对保留点熟)。
跟据树的性质,一个点有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;
}