[HDU 1011] Starship Troopers 树形dp

来源：互联网发布：十分钟学会python 编辑：程序博客网时间：2024/06/02 01:57

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;}

0 0