hdu1561(树形DP入门)

The more, The Better

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


 

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个城堡所获得的最多宝物的数量。

扫描二维码关注公众号,回复: 2596394 查看本文章

 

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

 

详细见注释

#include <bits/stdc++.h>
using namespace std;
const int N = 205;

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

vector<int> G[N];


void dfs(int s)
{
    for(int i = 0;i < G[s].size();i ++)
    {
        int son = G[s][i];

        if(G[son].size())  //如果还有延深  就继续向下搜
            dfs(son);
        
        //这里有背包的思想,从m到1  j遍历所有结点,下一个循环k遍历每个j的子结点  01背包  取或者不取  找最优
        for(int j = m;j >= 1;j --)
        {
            for(int k = 1;k < j;k ++)
                dp[s][j] = max(dp[s][j], dp[s][j - k] + dp[son][k]);
        }
    }
}

int main()
{

    while(cin >> n >> m && n && m)
    {
        //森林变树  把0结点当作总根
        m ++;

        memset(dp,0,sizeof(dp));

        for(int i = 0;i <= n;i ++) //初始化
            G[i].clear();

        int a, b;       //a相当于第i个结点的父节点  b是第i个结点的价值
        for(int i = 1;i <= n;i ++)
        {
            cin >> a >> b;
            G[a].push_back(i);//i作为a的儿子  加入表中
            
            // dp[i][j]  表示以第i个结点为根结点 选取j个结点的最大价值  首先都应该初始化为i结点本身的价值
            for(int j = 1;j <= m;j ++)
                dp[i][j] = b;

        }

        dfs(0);  //从根开始向下搜

        cout << dp[0][m] << endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Soul_97/article/details/81459863