多重背包问题 III(13-13)

107 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 13 天,点击查看活动详情

今天进行多重背包模型最后一题。

多重背包 III

题目描述

有 NN 种物品和一个容量是 VV 的背包。

第 ii 种物品最多有 sis_i 件,每件体积是 viv_i,价值是 wiw_i

求解将哪些物品装入背包,可使物品体积总和不超过背包容量,且价值总和最大。
输出最大价值。

输入格式

第一行两个整数,N,VN,V,用空格隔开,分别表示物品种数和背包容积。

接下来有 NN 行,每行三个整数 vi,wi,siv_i,w_i,s_i,用空格隔开,分别表示第 ii 种物品的体积、价值和数量。

输出格式

输出一个整数,表示最大价值。

数据范围

0<N1000\pmb{0<N≤1000}

0<V20000\pmb{0<V\le 20000}

0<vi,wi,si10000\pmb{0<v_i,w_i,s_i≤10000}

输入样例

4 5
1 2 3
2 4 1
3 4 3
4 5 2

输出样例:

10

题目分析

本题是楼天城 "男人八题" 中的一道题目,解法为单调队列优化多重背包

相较于昨天的多重背包题目,本题数据范围再次扩大,即使是 O(nvlogs)O(nvlogs) 的算法也无法通过全部样例。

首先我们将多重背包的转移方程列出,f[i][j]f[i][j] 表示从前 ii 件物品中选,总体积不大于 jj 的最优解。

{f[i,j]=max(f[i1,j],  f[i1,jv]+w,  f[i1,j2v]+2w,  ,f[i1,jsv]+sw)f[i,jv]=max(f[i1,jv],  f[i1,j2v]+w,  ,f[i1,jvsv]+sw)f[i,r+2v]=max(f[i1,r+2v],  f[i1,r+v]+w,  f[i1,r]+2w)f[i,r+v]=max(f[i1,r+v],  f[i1,r]+w,)f[i,r]=f[i1,r]\begin{cases} f[i,j]=max(f[i-1,j],\;f[i-1,j-v]+w,\;f[i-1,j-2v]+2w,\;\dots ,f[i-1,j-sv]+sw) \\f[i,j-v]=max(f[i-1,j-v],\;f[i-1,j-2v]+w,\;\dots ,f[i-1,j-v-sv]+sw) \\\dots \\f[i,r+2v]=max(f[i-1,r+2v],\;f[i-1,r+v]+w,\;f[i-1,r]+2w) \\f[i,r+v]=max(f[i-1,r+v],\;f[i-1,r]+w,) \\f[i,r]=f[i-1,r] \end{cases}

其中 r=j  mod  vr=j\; \text{mod} \;v

经过观察可以发现,在不考虑偏移量 ww 的情况下,f[i,j]f[i,j] 是数量为 ss 的相邻 f[i1,j],f[i1,jv]f[i1,jsw]f[i-1,j],f[i-1,j-v]\dots f[i-1,j-sw] 中的最大值。

于是可以使用滑动窗口进行优化。

考虑到偏移量 ww 的存在,由于其随着 f[i1,j]f[i-1,j]jj 距离初始点 rr 距离越近所加的 ww 越大,且每相隔一个 vv 便增加一个 ww,而我们仅考虑目标项中的最大值的下标而不需知道具体数值,则可以采取变加为减的方法,每距离 rr 多一个 vv 便多减一个 ww

然后我们枚举对于物品 ii 的体积 jj 的所有余数 0j10\sim j-1,便可得到 f[i,j]f[i,j]

如果尝试一维数组,发现每次枚举余数会已经改变 f[i,j]  (即下一个余数的f[i1][j])f[i,j]\;(即下一个余数的f[i-1][j]) 的值,所以需要多开一维数组存第 i1i-1 层的值。

最终复杂度可以控制到 O(NV)O(NV)

Accept代码 O(NV)

#include <bits/stdc++.h>

using namespace std;

const int N = 20010;
int f[N], g[N], q[N];

int main()
{
    int n, m;
    cin >> n >> m;
    
    for (int i = 0; i < n; i ++)
    {
        int v, w, s;
        cin >> v >> w >> s;
        memcpy(g, f, sizeof f);
        for (int j = 0; j < v; j ++)
        {
            int hh = 0, tt = -1;
            for (int k = j; k <= m; k += v)
            {
                while (hh <= tt && k - q[hh] > s * v) hh ++;
                while (hh <= tt && g[q[tt]] - q[tt] / v * w <= g[k] - k / v * w) tt --;
                q[++ tt] = k;
                f[k] = g[q[hh]] + (k - q[hh]) / v * w;
            }
        }
    }
    cout << f[m];
    return 0;
}