暑假笔记·背包进阶

分组背包

标准格式：

for 所有的组k
    for v = V ... 0
        for 所有的i属于组k
            f[v]=max{f[v],f[v-c[i]]+w[i]}

表述不清，可以看例题代码。

例题：P1064 金明的预算方案

原题地址

题目描述

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

主件 附件

电脑 打印机，扫描仪

书柜 图书

书桌 台灯，文具

工作椅 无

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

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

输入输出格式

输入格式：
第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是所属主件的编号）

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

输入输出样例

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

代码：
详细讲解
本题先将一个主件及其附件的不同组合看作不可共存的几个背包，然后运用分组背包解决。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<cmath>
#include<iostream>
#include<map>
#include<queue>
#include<string>
#include<vector>
using namespace std;
int n,m;
struct nod
{
    int p,val;
}qq[66],tg;
int f[32032],book[66],ans=0;
vector<int>mm;//主件的集合
vector<nod>aa[66],ss[66];//a[i]为第i个主件的附件集，ss[i]为第i个主件对应的不同背包
int main()
{
    int i,j,k,a,b,c,g;
    scanf("%d%d",&n,&m);
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        qq[i].p=a;qq[i].val=b;
        if(c)
        {
            aa[c].push_back(qq[i]);
        }
        else 
        {mm.push_back(i);book[i]=1;}
    }
    for(i=0;i<mm.size();i++)
    {
        g=mm[i];
        for(j=0;j<=n;j++) f[j]=-1;
        f[0]=0;//预处理
        for(j=0;j<aa[g].size();j++)//在不加主件的情况下对所有附件做个背包处理
        {
            for(k=n-qq[g].p;k>=aa[g][j].p;k--)
            {
                if(f[k]<f[k-aa[g][j].p]+aa[g][j].p*aa[g][j].val&&f[k-aa[g][j].p]!=-1) f[k]=f[k-aa[g][j].p]+aa[g][j].p*aa[g][j].val;
            }
        }
        for(j=1;j<=n-qq[g].p;j++)//统计共有多少种不同背包
        {
            if(f[j]!=-1)
            {
                tg.p=j+qq[g].p;tg.val=f[j]+qq[g].val*qq[g].p;
                ss[g].push_back(tg);
            }
        }
        tg.p=qq[g].p;tg.val=qq[g].val*qq[g].p;
        ss[g].push_back(tg);//不加附件只含主件
    }
    for(j=0;j<=n;j++) f[j]=-1;
    f[0]=0;//预处理
    for(i=0;i<mm.size();i++)//分组背包部分
    {
        g=mm[i];
        for(j=n;j>=0;j--)
        {
            for(k=0;k<ss[g].size();k++)//所有属于i组的k
            {
                if(j<ss[g][k].p) continue;
                if(f[j]<f[j-ss[g][k].p]+ss[g][k].val&&f[j-ss[g][k].p]!=-1) f[j]=f[j-ss[g][k].p]+ss[g][k].val;
            }
        }
    }
    for(i=1;i<=n;i++)
        ans=max(ans,f[i]);
    printf("%d\n",ans);
    return 0;
}