HDU1561-The more, The Better（树形DP 树形依赖背包）

题目链接

Problem Description

ACboy很喜欢玩一种战略游戏，在一个地图上，有N座城堡，每座城堡都有一定的宝物，在每次游戏中ACboy允许攻克M个城堡并获得里面的宝物。但由于地理位置原因，有些城堡不能直接攻克，要攻克这些城堡必须先攻克其他某一个特定的城堡。你能帮ACboy算出要获得尽量多的宝物应该攻克哪M个城堡吗？

Input

每个测试实例首先包括2个整数，N,M.(1 <= M <= N <= 200);在接下来的N行里，每行包括2个整数，a,b. 在第 i 行，a 代表要攻克第 i 个城堡必须先攻克第 a 个城堡，如果 a = 0 则代表可以直接攻克第 i 个城堡。b 代表第 i 个城堡的宝物数量, b >= 0。当N = 0, M = 0输入结束。

Output

对于每个测试实例，输出一个整数，代表ACboy攻克M个城堡所获得的最多宝物的数量。

Sample Input

3 2

0 1

0 2

0 3

7 4

2 2

0 1

0 4

2 1

7 1

7 6

2 2

0 0

Sample Output

5

13

---

题意

类似于背包问题，有n个物品，可以拿m个物品，每个物品都有一个价值。但是拿的顺序有要求，如果B是A的子结点，那么必须先拿A，然后才能拿B。即有依赖的背包问题，这个依赖是树形结构。

思路

树形DP，dp[ i ] [ j ] 表示第 i 个结点拿 j 个物品时最大的价值。

树就用DFS走一遍，状态转移方程是dp[now][j] = max(dp[now][j],dp[now][k]+dp[u][j-k]);

 第一个for循环的 j 表示走的步数，必须是倒序，类似于01背包滚动数组。

第二个for循环的 k 表示当前根结点走的步数，j - k是子结点走的步数，（走的步数在这道题目中就是攻克的城堡数量，在背包中就是拿的物品的数量）。

按照第一个图，走到了第四个结点，这时候的dp [ now ][ j ] 其实不是所有的结点，而只是前三个、已经走过的结点的最大值。这时候想要把第四个结点u添加。求dp[ now ] [ 5 ] （假如m是5，即最大的步数），可以在前三个（now结点）和第四个子结点（u结点）比较，分别走1-4，2-3，3-2，4-1步，这样就列举了所有的情况。这时候的最大值dp[ now ] [ 5 ]，是前四个结点走五步的最大值。然后算dp[ now ] [ 4 ] ，也是同样的方法，now和u结点分别走1-3，2-2，3-1步，求出最大的就是dp[ now ] [ 4 ]。这时候就显示出为什么要倒着计算了。因为如果正序计算，dp[ now ] [ 1 ]已经变成了前四个点的最大值，在求dp[ now ] [ 2 ]就没有办法计算了，因为dp[ now ] [ 1 ]已经被更新成前四个结点，而不是前三个结点的最大值。

还有就是题目给的数据是森林，可以人为的定义0为根节点，这样就成了一棵树，步数就要走m+1步，因为多走了根节点。

---

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;

const int maxn = 210;

struct Edge
{
    int to,next;
}edge[maxn*2];

int head[maxn],tot=0;
void addEdge(int from,int to)
{
    edge[tot].to = to; edge[tot].next = head[from];
    head[from] = tot++;
}

int dp[maxn][maxn],n,m;

void dfs(int now,int fa)
{
    for(int i=head[now];i!=-1;i=edge[i].next){
        int u = edge[i].to;
        if(u==fa) continue;
        dfs(u,now);
        for(int j=m;j>1;j--){
            for(int k=1;k<j;k++){
                dp[now][j] = max(dp[now][j],dp[now][k]+dp[u][j-k]);
            }
        }
    }
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        if(m==0&&n==0) break;
        m++;//0为根，要多走1步
        tot=0;
        memset(head,-1,sizeof(head));
        memset(edge,0,sizeof(edge));
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++){
            int x,y; scanf("%d%d",&x,&y);
            addEdge(x,i);
            for(int j=1;j<=m;j++)
                dp[i][j] = y;
        }
        dfs(0,-1);
        printf("%d\n",dp[0][m]);

    }
    return 0;
}