「BSOJ2163」 人品问题 - 树形Dp

题目大意

在一棵以1为根,各节点编号在1~n的二叉树上,选出1以外的k个节点,使得选出来的点的权值和最大,并且满足除1的两个儿子以外的节点的父亲都被选上。输出最大的权值和。

分析

比较水的树型DP。

f[i][j]为以i为根的子树选j个节点的最大权值和,则f[i][j]=\max_{0\le t<j}\{f[lc[i]][t]+f[rc[i][j-t-1]]+v[i]\}。再注意一下细节就可以了。

代码

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

 

你可能感兴趣的:(动态规划---树形Dp)