树形依赖背包模板

最新推荐文章于 2022-09-02 13:27:20 发布

菜菜魁

最新推荐文章于 2022-09-02 13:27:20 发布

阅读量334

点赞数 1

分类专栏：动态规划文章标签：树形依赖背包金明的预算方案

本文链接：https://blog.csdn.net/qq_38367681/article/details/80139518

版权

动态规划专栏收录该内容

12 篇文章 1 订阅

订阅专栏

金明的预算方案

金明今天很开心，家里购置的新房就要领钥匙了，新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是，妈妈昨天对他说：“你的房间需要购买哪些物品，怎么布置，你说了算，只要不超过N元钱就行”。今天一早，金明就开始做预算了，他把想买的物品分为两类：主件与附件，附件是从属于某个主件的，下表就是一些主件与附件的例子：

主件	附件
电脑	打印机，扫描仪
书柜	图书
书桌	台灯，文具
工作椅	无

</dd></dl>

如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有0个、1个或2个附件。附件不再有从属于自己的附件。金明想买的东西很多，肯定会超过妈妈限定的N元。于是，他把每件物品规定了一个重要度，分为5等：用整数1~5表示，第5等最重要。他还从因特网上查到了每件物品的价格（都是10元的整数倍）。他希望在不超过N元（可以等于N元）的前提下，使每件物品的价格与重要度的乘积的总和最大。

设第j件物品的价格为v[j]，重要度为w[j]，共选中了k件物品，编号依次为j1，j2，……，jk，则所求的总和为：

v[j1]*w[j1]+v[j2]*w[j2]+ …+v[jk]*w[jk]。（其中*为乘号）

请你帮助金明设计一个满足要求的购物单。

输入描述 Input Description

第1行，为两个正整数，用一个空格隔开：

N m

（其中N（<32000）表示总钱数，m（<60）为希望购买物品的个数。）

从第2行到第m+1行，第j行给出了编号为j-1的物品的基本数据，每行有3个非负整数

v p q

（其中v表示该物品的价格（v<10000），p表示该物品的重要度（1~5），q表示该物品是主件还是附件。如果q=0，表示该物品为主件，如果q>0，表示该物品为附件，q是所属主件的编号）

输出描述 Output Description

只有一个正整数，为不超过总钱数的物品的价格与重要度乘积的总和的最大值（<200000）

样例输入 Sample Input

1000 5

800 2 0

400 5 1

300 5 1

400 3 0

500 2 0

样例输出 Sample Output

2200

这种背包并不是简单的选取最优，有些会遇到一些限制条件，比如选a必须先选b，那怎么办，这里需要用到dfs序，dfs序是个很神奇的东西，大家都知道深搜的过程，在回溯时，它一定是由儿子节点回溯到父亲节点，这样的话，我们把a当成b的儿子节点，选儿子时，父亲必选，这样的背包就可以解决问题。

#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
vector<int >mp[250];
struct l{
    int w, v;
}a[250];
int dp[250][32010];
int vis[250];
int n,m;
void Dfs(int now, int m)
{
    vis[now]=1;
    for(int i=0;i<mp[now].size();i++)
    {
        int to=mp[now][i];
        for ( int j = 0; j <= m - a[to].w; ++j )
        {
            dp[to][j] = dp[now][j];
        }
        if(vis[to]==0 && m - a[to].w > 0)
        {
            Dfs(to, m - a[to].w);
        }
        for(int j=m;j >= a[to].w;j--)
        {
            dp[now][j] = max(dp[now][j], dp[to][j - a[to].w] + a[to].v);
        }
    }
}
int main()
{
    scanf("%d%d",&m,&n);
    memset(vis,0,sizeof(vis));
    memset(dp,0,sizeof(dp));
    for(int i=0;i<=n;i++)mp[i].clear();
    for(int i=1;i<=n;i++)
    {
        int w,x,val;
        scanf("%d%d%d",&w, &val, &x);
        mp[x].push_back(i);
        a[i].w = w;
        a[i].v = w * val;
    }
    Dfs(0, m);
    int ans = 0;
    for(int i = 0; i <= m; i++)
        ans = max(ans, dp[0][i]);
    printf("%d\n",ans);return 0；
}