多重背包单调队列优化详解

6. 多重背包问题 III - AcWing

分析：

• 最暴力的写法 $O(NVS)，4e11$ , 显然会超时

  （这是 4. 多重背包问题 I - AcWing 的写法（见文末））

• 本题要通过单调队列将时间复杂度优化到 $O(NV)$, 优化掉一个 $S$

• 为什么能这样优化，先看一下原始的状态转移方程：

  f[j]=max(f[j],f[j-v]+w,f[j-2v]+2w,...,f[j-sv]+sw)
  也就是说，f[j]是由f[0~j]状态中与 j 相差 k*v 转移过来的
  考虑正推，打个表：
      f[j]    = f[j]
      f[j+v]  = max(f[j]+w,f[j+v])
      f[j+2v] = max(f[j]+2w,f[j+v]+W,f[j+2v])
      f[j+3v] = max(f[j]+3w,f[j+v]+2W,f[j+2v]+w,f[j+3v])
      f[j+4v] = max(f[j]+4w,f[j+v]+3W,f[j+2v]+2w,f[j+3v]+w,f[j+4v])
      ... ...
  

• 通过上表发现：要得到 $f[j+kv]$ 的最大值，只需要维护 $(f[j]+kw,f[j+v]+(k-1)w,...,f[j+(k-1)v]+w)$ 的最大值即可

• 怎么用单调队列去维护过程中的最大值，先将框架打出来

  for(int i=1;i<=n;i++)
  {
      int v,w,s;
      cin>>v>>w>>s;
      memcpy(pre,f,sizeof(f));
      for(int j=0;j<v;j++)
      {
          int head=0,tail=-1;
          for(int k=j;k<=m;k+=v)
          {
              
          }
      }
  }
  

• 可以想到 $q[]$ 里存的是 $k(即下标j,j+v,j+2v,...,j+kv)$

• 也容易想到队首出队的条件

  if(head<=tail && k-s*v>q[head]) head++
  

• 最麻烦的就是队尾出队的条件，因为各个 $f[k]$ 加的 $w$ 个数不统一，这里要有一个巧妙的转换

  f[j+4v] = max(f[j]+4w,f[j+v]+3W,f[j+2v]+2w,f[j+3v]+w,f[j+4v])
  f[j+4v] = max(f[j],f[j+v]-w,f[j+2v]-2w,f[j+3v]-3w,f[j+4v]-4w)+4w
  

  发现 $f[j+kv]-kw$

  这样一来，队尾出队条件就出来了：

  while(head<=tail && pre[q[tail]]-(q[tail]-j)/v*w<=pre[k]-(k-j)/v*w) tail--;
  

总结：

• 第一个坑：想到用类似筛法（倍增），通过 $k+=v$ 来枚举所有状态（可以看一下196. 质数距离 - AcWing题库（素数筛，区间，筛两遍）_yezzz的博客-CSDN博客）
• 第二个坑：想到往单调队列靠
• 第三个坑：想到那步巧妙的加法变成减法

#include <bits/stdc++.h>
using namespace std;

const int M=20005;
int f[M],pre[M]; // 用pre存取f[i-1][]的所有状态
int q[M];
signed main()
{
    ios_base::sync_with_stdio(0);
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int v,w,s;
        cin>>v>>w>>s;
        memcpy(pre,f,sizeof(f));
        for(int j=0;j<v;j++)
        {
            int head=0,tail=-1;
            for(int k=j;k<=m;k+=v)
            {
                if(head<=tail && k-s*v>q[head]) head++;
                if(head<=tail) f[k]=max(f[k],pre[q[head]]+(k-q[head])/v*w);
                while(head<=tail && pre[q[tail]]-(q[tail]-j)/v*w<=pre[k]-(k-j)/v*w) tail--;
                q[++tail]=k;
            }
        }
    }
    cout<<f[m]<<endl;
	return 0;
}

4. 多重背包问题 I - AcWing

• $O(NVS)$

#include <bits/stdc++.h>
using namespace std;

const int N=1005;
int f[N];
signed main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int v,w,s;
        cin>>v>>w>>s;
        for(int j=m;j>=v;j--)
        {
            for(int k=1;k<=s && k*v<=j;k++)
            {
                f[j]=max(f[j],f[j-k*v]+k*w);
            }
        }
    }
    cout<<f[m]<<endl;
	return 0;
}