DP - 分组背包(有依赖的背包问题) - NOIP2006提高组 - 金明的预算方案

DP - 分组背包(有依赖的背包问题) - NOIP2006提高组 - 金明的预算方案

金明今天很开心，家里购置的新房就要领钥匙了，新房里有一间金明自己专用的很宽敞的房间。

更让他高兴的是，妈妈昨天对他说：“你的房间需要购买哪些物品，怎么布置，你说了算，只要不超过N元钱就行”。

今天一早，金明就开始做预算了，他把想买的物品分为两类：主件与附件，附件是从属于某个主件的，下表就是一些主件与附件的例子：

如果要买归类为附件的物品，必须先买该附件所属的主件。

每个主件可以有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]（其中*为乘号）

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

输入格式
输入文件的第1行，为两个正整数，用一个空格隔开：N m，其中N表示总钱数，m为希望购买物品的个数。

从第2行到第m+1行，第j行给出了编号为j-1的物品的基本数据，每行有3个非负整数v p q，其中v表示该物品的价格，p表示该物品的重要度（1~5），q表示该物品是主件还是附件。

如果q=0，表示该物品为主件，如果q>0，表示该物品为附件，q是所属主件的编号。

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

数据范围
N<32000,m<60,v<10000

输入样例：
1000 5
800 2 0
400 5 1
300 5 1
400 3 0
500 2 0
输出样例：
2200

---

分析：

$将主件与附件视作一个整体，由于附件数量有限(<=2)，因此整体的状态有4种：$

$$\begin{gathered} ①、主件+0个附件，表示为00 \\ ②、主件+附件1，表示为01 \\ ③、主件+附件2，表示为10 \\ ④、主件+2个附件，表示为11 \end{gathered}$$

$这样，我们将每个整体存储下来以后，对整体跑一遍分组背包即可。$

具体落实：

$用pair存储主件，包括主件价格(物品体积)，主件价值(价格\times 重要度)。$

$vector<pair>存附件，下标为依附的主件编号，元素的值与主件包含的值对应。$

代码：

#include<iostream>
#include<algorithm>
#include<vector>

#define P pair<int,int>
#define v first
#define w second

using namespace std;

const int N=32010;

int m,n,f[N];
P master[N];
vector<P> serve[N];

int main()
{
    cin>>m>>n;
    for(int i=1;i<=n;i++)
    {
        int v,p,q;
        cin>>v>>p>>q;
        if(!q) master[i]={v,v*p};
        else serve[q].push_back({v,v*p});
    }
    
    for(int i=1;i<=n;i++)   //物品总数
        for(int j=m;j>=0;j--)   //背包容量
        {
            for(int k=0; k < 1<<serve[i].size(); k++)   //每组物品数量
            {
                int v=master[i].v,w=master[i].w;
                for(int u=0;u<serve[i].size();u++)   //具体方案
                    if(k>>u & 1)
                    {
                        v+=serve[i][u].v;
                        w+=serve[i][u].w;
                    }
                if(j>=v) f[j]=max(f[j],f[j-v]+w);
            }
        }
        
    cout<<f[m]<<endl;
    
    return 0;
}