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