题目大意
在一棵以1为根,各节点编号在1~n的二叉树上,选出1以外的k个节点,使得选出来的点的权值和最大,并且满足除1的两个儿子以外的节点的父亲都被选上。输出最大的权值和。
分析
比较水的树型DP。
设为以为根的子树选个节点的最大权值和,则。再注意一下细节就可以了。
代码
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
int n,k;
int lc[105],rc[105],val[105];
int f[105][105];
bool vis[105][105];
int Dp(int num,int cnt)
{
if (vis[num][cnt]) return f[num][cnt];
if (cnt==0) return f[num][cnt]=0;
if (num==0) return f[num][cnt]=-0x3f3f3f3f;
if (cnt==1) return f[num][cnt]=val[num];
f[num][cnt]=-0x7fffffff/2;
for (int i=0;i<=cnt-1;i++)
f[num][cnt]=max(f[num][cnt],val[num]+Dp(lc[num],i)+Dp(rc[num],cnt-i-1));
vis[num][cnt]=1;
return f[num][cnt];
}
int main()
{
scanf("%d%d",&n,&k);
for (int i=2;i<=n;i++)
scanf("%d",&val[i]);
for (int i=1;i<=n;i++)
scanf("%d%d",&lc[i],&rc[i]);
printf("%d",Dp(1,k+1));
return 0;
}