【树形dp】JZOJ_3914 人品问题

题意

给出一棵树。一个节点至多有两个祖先，如果选了某个祖先，那么它的后代必须要选一个。每个点上有一个权值，求出从$1$号节点取$K$个祖先的最大总权值。

思路

很明显的树形动态规划，数据较小，这里打了$O(N^3)$的做法。

设$f[i][j]$为第$i$个点选了$j$个祖先的最大权值，转移显然。

答案就为$f[1][k]$。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>

int n, k;
int fa[101][2], sum[101];
long long v[101], f[101][101];

void calcTree(int p) {
	if (!p) return;
	sum[p] = 1;
	if (fa[p][0]) {
		calcTree(fa[p][0]);
		sum[p] += sum[fa[p][0]];
	}
	if (fa[p][1]) {
		calcTree(fa[p][1]);
		sum[p] += sum[fa[p][1]];
	}
}

void dp(int p) {
	if (!p) return;
	dp(fa[p][0]);
	dp(fa[p][1]);
	f[p][0] = v[p];
	for (int i = 0; i < sum[p]; i++) {
		if (sum[fa[p][0]] - 1 >= i)
			f[p][i + 1] = std::max(f[p][i + 1], f[fa[p][0]][i] + v[p]);
		if (sum[fa[p][1]] - 1 >= i)
			f[p][i + 1] = std::max(f[p][i + 1], f[fa[p][1]][i] + v[p]);//选任意一个
		for (int j = 0; j <= i; j++)
			if (sum[fa[p][0]] - 1 >= j && sum[fa[p][1]] - 1 >= i - j)
				f[p][i + 2] = std::max(f[p][i + 2], f[fa[p][0]][j] + f[fa[p][1]][i - j] + v[p]);//两个祖先都选
	}
}

int main() {
	memset(f, -127 / 3, sizeof(f));
	scanf("%d %d", &n, &k);
	for (int i = 1; i < n; i++)
		scanf("%lld", &v[i + 1]);
	for (int i = 1; i <= n; i++)
		scanf("%d %d", &fa[i][0], &fa[i][1]);
	calcTree(1);
	dp(1);
	printf("%lld", f[1][k]);
}