题意:有n个洞组成一棵树形结构,你有m个士兵,你从1号房间开始攻打,每个洞有a个bugs和brain的价值。你的一个士兵可以打20个bugs,为了拿到这个洞的价值brain你必须留下k个士兵消灭这个洞的所有bugs(k*20>="bugs"的数量,且留下的士兵不可以再去攻打其他的洞,且必须攻打了前面的洞才可以攻打后面的洞)。问你花费这m个士兵可以得到的最大价值是多少。
思路:树形dp dp[i][j]表示第i个节点安排j个士兵所能取得的最大价值,状态转移方程:
dp[dad][j+k] = max(dp[dad][j+k], dp[dad][j] + dp[son][k]); 其中m = i + j;
ps:提议并未给出明确的谁父谁子,所以建边时 要建双向边。
#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <cstdio>
#include <string>
#include <bitset>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <list>
#include <map>
#include <set>
#define sss(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a) memset(a,0,sizeof(a))
#define ss(a,b) scanf("%d%d",&a,&b)
#define s(a) scanf("%d",&a)
#define INF 0x3f3f3f3f
#define w(a) while(a)
#define PI acos(-1.0)
#define LL long long
#define eps 10E-9
#define N 100010<<1
#define mod 1000000000+7
using namespace std;
void mys(int& res)
{
int flag=0;
char ch;
while(!(((ch=getchar())>='0'&&ch<='9')||ch=='-'))
if(ch==EOF) res=INF;
if(ch=='-') flag=1;
else if(ch>='0'&&ch<='9') res=ch-'0';
while((ch=getchar())>='0'&&ch<='9') res=res*10+ch-'0';
res=flag?-res:res;
}
void myp(int a)
{
if(a>9)
myp(a/10);
putchar(a%10+'0');
}
/*************************THE END OF TEMPLATE************************/
struct node {
int bugs;
int brains;
int pos;
int son[101];
}info[110];
int dp[110][110];
bool vis[110];
int n, m;
void tree_dp(int root){
vis[root] = true;
int cost = (info[root].bugs + 19)/20;
for(int i = cost; i<=m; i++) dp[root][i] = info[root].brains;
for(int i = 0; i<info[root].pos; i++){
int next = info[root].son[i];
if(!vis[next]){
tree_dp(next);
for(int j = m; j>=cost; j--){
for(int k = 1; j+k<=m; k++){
if(dp[next][k]){//当前情况肯定要在子情况已经明确的情况下决定
dp[root][j + k] = max(dp[root][j+k], dp[root][j] + dp[next][k]);
}
}
}
}
}
}
int main(){
w(~ss(n, m)){
if(n == -1 && m == -1) break;
mem(info);
mem(dp);
mem(vis);
for(int i=1; i<=n; i++) ss(info[i].bugs, info[i].brains);
int x, y;
for(int i = 1; i < n; i++){
ss(x, y);
info[x].son[info[x].pos++] = y;
info[y].son[info[y].pos++] = x;
}
if(!m) printf("0\n");
else{
tree_dp(1);
printf("%d\n", dp[1][m]);
}
}
return 0;
}