[HDU 1011] Starship Troopers 树形dp

最新推荐文章于 2024-10-16 16:55:34 发布

原创最新推荐文章于 2024-10-16 16:55:34 发布 · 608 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#ACM #HDU #dp

动态规划专栏收录该内容

25 篇文章

订阅专栏

本文详细介绍了如何使用树形dp解决HDOJ平台上的1011题，该题涉及利用树结构进行动态规划求解，通过转移方程dp[i][k]=max(dp[i][k],dp[i][k-l]+dp[i][l])来实现最优解的计算。

http://acm.hdu.edu.cn/showproblem.php?pid=1011

题意：有一支部队，每个人能消灭 20 个 bug。输入n , m。表示 bug 有 n 个洞（编号1-n ）和这支部队有 m 个人。这些洞形成一颗树，必须打了前面才能打到后面。

思路：树形dp，转移方程：
dp[i][k] = max(dp[i][k], dp[i][k-l] + dp[i][l]);

#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

int n, m;
int dp[105][105];
vector<int> eg[105];
int vol[105],val[105];

void Dfs(int rt, int pre)
{
    int len = eg[rt].size();
    for(int i = vol[rt]; i <= m; i++){
        dp[rt][i] = val[rt];
    }
    for(int i = 0; i < len; i++) 
    {
        int son = eg[rt][i];
        if(pre == son)
            continue;
        Dfs(son, rt);
        for(int k = m; k >= vol[rt]; k--) //必须打了rt 才能打son，所以k > vol[rt]
        {
             for(int l = 1; l <= k - vol[rt]; l++) //派 l 个人去打 son
             {
                 if(dp[rt][k] < dp[rt][k-l] + dp[son][l])
                    dp[rt][k] = dp[rt][k-l] + dp[son][l];
             }
        }
    }
}

int main()
{
    while(cin>>n>>m && (m != -1 || n != -1)){
        memset(dp, 0, sizeof(dp));
        for(int i = 1; i <= n; i++){
            cin>>vol[i]>>val[i];
            vol[i] = (vol[i] / 20) + (vol[i] % 20 != 0);
            eg[i].clear();
        }
        int x, y;
        for(int i = 1; i < n; i++){
            cin>>x>>y;
            eg[x].push_back(y);
            eg[y].push_back(x);
        }
        if(m == 0){
            cout<<"0"<<endl;
            continue;
        }
        Dfs(1, -1);
        cout<<dp[1][m]<<endl;
    }
    return 0;
}