「BSOJ2163」人品问题 - 树形Dp

TbYangZ

于 2018-08-23 18:45:01 发布

阅读量234

点赞数

分类专栏：动态规划---树形Dp 文章标签：树型dp

本文链接：https://blog.csdn.net/sin_Yang/article/details/81985959

版权

动态规划---树形Dp 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

题目大意

在一棵以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 <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;
}