hdu1011 树形dp

来源：互联网发布：淘宝回购率编辑：程序博客网时间：2024/05/16 09:23

题意：有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+7using 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;}

1 0