hdu 1561 The more, The Better 树形dp+背包

The more, The Better

Time Limit: 6000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 5385    Accepted Submission(s): 3195

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
  
  

 

Author

8600

 

Source

HDU 2006-12 Programming Contest

 

Recommend

LL   |   We have carefully selected several similar problems for you:   1011  2159  1203  2602  1712 

 

思路：树形dp，但这题可能有多棵树，要分别处理每棵树，用root数组记录每棵树的根节点，最后dp出结果。

这题有限制条件，针对每个节点，如果不取这个节点，那么它的子节点也不可取，若取子节点，父节点必取。

所以先对子节点背包，最后加上父节点，背包的时候先直接考虑父节点会造成父节点收益计算重复。

dp[i][j]表示到i节点时取j个点的最大收益。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <queue>
#include <map>
using namespace std;
const int N=205;
const int inf=0x3f3f3f3f;
struct node
{
    int v,next;
}edge[N];
int p1=1,p2=1;
int n,m,a,b;
int dp[N][N],head[N],v[N];
int root[N];
void addedge(int i,int j)
{
    edge[p1].v=j;
    edge[p1].next=head[i];
    head[i]=p1++;
}
void init(){
    p1=1,p2=1;
    memset(dp,0,sizeof(dp));
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++){
        scanf("%d%d",&a,&b);
        if(a==0)root[p2++]=i;
        else addedge(a,i);
        v[i]=b;
    }
}
void dfs(int u){
    if(head[u]==-1){
        for(int i=1;i<=m;i++)dp[u][i]=v[u];
        return;
    }
    for(int i=head[u];i!=-1;i=edge[i].next){
        int vt=edge[i].v;
        dfs(vt);
        for(int j=m-1;j>=0;j--){
            for(int k=0;k<=j;k++){
            dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[vt][k]);
            }
        }
    }
    for(int i=m;i>0;i--)dp[u][i]=dp[u][i-1]+v[u];
}
int main()
{
    while(scanf("%d%d",&n,&m),n||m){
        init();
        for(int i=1;i<p2;i++)dfs(root[i]);
        for(int i=1;i<p2;i++){
            for(int j=m;j>=0;j--){
                for(int k=0;k<=j;k++){
                    dp[0][j]=max(dp[0][j],dp[0][j-k]+dp[root[i]][k]);
                }
            }
        }
        printf("%d\n",dp[0][m]);
    }
    return 0;
}