hdu 1561 The more, The Better 树形DP入门题

来源：互联网发布：linux 文件删除编辑：程序博客网时间：2024/05/03 02:23

题意：中文题，简单明了。

解题思路：首先将所有的可以直接攻克的城堡并在节点0的下面，然后对于每个节点，利用分组背包的思想，该节点的各个子树看成各个背包。

#include <stdio.h>#include <string.h>struct EDGE {int to, next;}edge[222];int head[222], E, cnt[222], dp[222][222], val[222];void newedge(int u, int to) {edge[E].to = to;edge[E].next = head[u];head[u] = E++;}void init() {memset(head, -1, sizeof(head));memset(dp, 0 , sizeof(dp));E = 0;}void dfs(int u) {cnt[u] = 1; if(head[u] == -1){dp[u][1] = val[u];return ;}int i, j, k;for(i = head[u];i != -1; i = edge[i].next) {int to = edge[i].to;dfs(to);cnt[u] += cnt[to];for(j = cnt[u]-1;j >= 0; j-- ) {for(k = 1;k <= cnt[to]; k++) if(j+k < cnt[u] && dp[u][j]+dp[to][k] > dp[u][j+k])dp[u][j+k] = dp[u][j] + dp[to][k];}}if(u != 0) {for(i = cnt[u];i >= 1; i--)dp[u][i] = dp[u][i-1]+val[u];dp[u][0] = 0;}}int main() {int i, j, k, n, m, a;while(scanf("%d%d", &n, &m) != -1 && n) {init();for(i = 1;i <= n; i++) {scanf("%d%d", &a, &val[i]);newedge(a, i);}val[0] = 0;dfs(0);printf("%d\n", dp[0][m]);}return 0;}