hdu1561 The more, The Better 树形dp+依赖背包

 题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1561

 

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

 

做这题前又看了一遍背包9讲，感觉太经典了，尤其是泛化背包，简直是精华。这题的关系就是裸地依赖背包，用树形dp解。

首先，限制条件是选择m个物品，而每个物品最多选一次，跟0-1背包的区别在于有依赖关系，那么这层依赖关系我们可以借助于一个树来解决。借助dfs，从根节点开始dfs，然后直到叶子节点，回朔的时候进行0-1背包dp。

定义状态：dp [ i ] [ j ] 表示在节点i，从以i为根节点的子树下选择j个城市的最大价值

初始化：dp [ i ] [ j ] =val [ i ]（i节点的价值）（1 < = j < = m）

转移方程 dp【father】【j】 = max （dp【father】【j】，dp【father】【k】+dp【child】【j-k】）；由前面的dfs可见，我们是用子节点更新父节点，用j枚举父节点选择的城市数，k枚举留给其他子节点选择城市数，那么就可以转移了

注意：此题出给很多初始节点，也就是有很多森林，我们要设置一个超级root连接子树的根节点即可。

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;

int n,m,x,y;
int dp[205][205]; //dp[i][j]表示在节点i，从以i为根节点的子树下选择j个城市的最大价值
vector<int> g[205];

void dfs(int u,int fa){

    for(int i=0;i<g[u].size();i++){
        int v = g[u][i];
        if(v==fa)
            continue;
        dfs(v,u);//没有fa,应该不会回头
        for(int j=m;j>1;j--)
            for(int k=1;k<j;k++)
                dp[u][j] = max(dp[u][j] ,dp[u][k]+dp[v][j-k]);
    }

    return;
}

int main(){

    while(~scanf("%d%d",&n,&m),n+m){
        m++;//建树+一个空root节点
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++){
            scanf("%d%d",&x,&y);
            g[x].push_back(i);//建图
            for(int j=1;j<=m;j++)
                dp[i][j] = y;
        }
        dfs(0,-1);
        printf("%d\n",dp[0][m]);
        for(int i=0;i<=n;i++)
            g[i].clear();
    }

    return 0;
}

另外一种风格

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;

int n,m,x,y;
int dp[205][205]; //dp[i][j]表示在节点i，从以i为根节点的子树下选择j个城市的最大价值
vector<int> g[205];

void dfs(int u,int m){

    for(int i=0;i<g[u].size();i++){
        int v = g[u][i];
        if(m>1)
            dfs(v,m-1);
        for(int j=m;j>1;j--)
            for(int k=1;k<j;k++)
                dp[u][j] = max(dp[u][j] ,dp[u][k]+dp[v][j-k]);
    }

    return;
}

int main(){

    while(~scanf("%d%d",&n,&m),n+m){
        m++;//建树+一个空root节点
        memset(dp,0,sizeof(dp));
        for(int i=0;i<=n;i++)
            g[i].clear();
        for(int i=1;i<=n;i++){
            scanf("%d%d",&x,&y);
            g[x].push_back(i);//建图
            for(int j=1;j<=m;j++)
                dp[i][j] = y;
        }
        dfs(0,m);
        printf("%d\n",dp[0][m]);

    }

	return 0;
}