hdu 1011 Starship Troopers 树形背包

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1011

题意:有n个洞,每个洞有若干虫子和脑子,洞之间相连形成一棵树。你有m个士兵,一个士兵可以打10只虫子,士兵留下后就不能走了。从一号洞开始打,求获得最多的脑子。


树形背包。如果你在某一个洞里,有m个士兵。你可以选择派遣k个士兵去攻打与洞相连的某个子树。可以先用递归算出子树的情况。这类似于背包九讲的泛化背包。

定义
考虑这样一种物品,它并没有固定的费用和价值,而是它的价值随着你分配给它的费用而变化。这就是泛化物品的概念。
更严格的定义之。在背包容量为V的背包问题中,泛化物品是一个定义域为0..V中的整数的函数h,当分配给它的费用为v时,能得到的价值就是h(v)

这里的背包容量就为此刻的士兵数。选择某个子树就为选择某个物品。而根据派遣的士兵数不同,返回的最大脑子数也不一样。

设d[t][j]为在第t个洞有j个士兵能得到的最大脑子数,c为一个子树,k为派遣给子树的士兵。所以有d[t][j]=max(d[t][j],d[t][j-k]+d[c][k]);

注意如果没有士兵就拿不到脑子,即使没有虫子。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#define N 110
using namespace std;

vector<int> r[N];
int d[N][N],brain[N],bugs[N],v[N],m;

void dfs(int t)
{
    v[t]=1;
    int m1=(bugs[t]+19)/20;
    for(int i=m1;i<=m;i++)  d[t][i]=brain[t];
    for(int i=0;i<r[t].size();i++)
    {
        int c=r[t][i];
        if(v[c])   continue;
        dfs(c);
        for(int j=m;j>=m1;j--)
            for(int k=1;k<=j-m1;k++)
                d[t][j]=max(d[t][j],d[t][j-k]+d[c][k]);
    }
}

int main()
{
    int n;
    while(~scanf("%d%d",&n,&m)&&(n!=-1||m!=-1))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&bugs[i],&brain[i]);
            r[i].clear();
        }
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            r[u].push_back(v);
            r[v].push_back(u);
        }
        if(m==0)
        {
            cout<<0<<endl;
            continue;
        }
        memset(d,0,sizeof(d));
        memset(v,0,sizeof(v));
        dfs(1);
        cout<<d[1][m]<<endl;
    }
}


  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值