mibbp

《男人八题》之—多重背包的单调队列优化与二进制优化

  多重背包就是背包问题加了个次数每个物品可以选s次,很容易想到的代码就是在状态转换中再加一个循环从0到s,但这样大数据会超时,所以有了很多

优化方案。

  二进制优化:

  每个物品可以选的次数都是不同的,有一种想法是把这可以选s次的物品拆成s个物品,这样就只涉及选与不选,就简化成了01背包问题但还是会超时,因

为拆成了s个但其实没必要我们只需要拆成log2(n)个就可以了,转换成二进制,举个例子,比如说7,7的二进制是111,111以内的数都可以用100,10,1来表示,也就是十进制的1,2,4,但7是特殊情况或者比如说10,同样拆成1、2、4。1、2、4,可以表示7以内的数而我想表示10以内的书就只需要再加个3就可以了,dddd。

  然后这样代码复杂度就不是O(n*v*s)而是O(n*v*log2(s))。

  代码:

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 2010;
int f[N],n,m;
struct good
{
    int w,v;
};

int main()
{
    cin>>n>>m;
    vector<good> Good;
    good tmp;
    for(int i = 1 ; i <= n ; i++ )
    {
        int v,w,s;
        cin>>v>>w>>s;
        for(int k = 1 ; k <= s ; k*=2 )
        {
            s-=k;
            Good.push_back({k*w,k*v});
        }
        if(s>0) Good.push_back({s*w,s*v});
    }
    for(auto t : Good)
        for(int j = m ; j >= t.v ; j--)
            f[j] = max(f[j] , f[j-t.v]+t.w );


    cout<<f[m]<<endl;
    return 0;

}

  单调队列优化:

  这个巨难我看大佬解析看懂的,原解析在这

  作者:lys
  链接:https://www.acwing.com/solution/content/6500/
  来源:AcWing

  首先先从完全背包的思路出发dp[i][j] 表示将前 i 种物品放入容量为 j 的背包中所得到的最大价值dp[i][j] = max(不放入物品 i,放入1个物品 i,放入2个物品 i, ... , 放入k个物品 i),这里 k 要满足:k <= s, j - k*v >= 0,不放物品  i = dp[i-1][j],放k个物品 i = dp[i-1][j - k*v] + k*w,dp[i][j] = max(dp[i-1][j], dp[i-1][j-v] + w, dp[i-1][j-2*v] + 2*w,..., dp[i-1][j-k*v] + k*w)。

  然后我们优化成了一维,适当的调整循环条件,我们可以重复利用dp数组来保存上一轮的信息,我们令 dp[j] 表示容量为j的情况下,获得的最大价值,那么,针对每一类物品 i ,我们都更新一下 dp[m] --> dp[0] 的值,最后 dp[m] 就是一个答案。
  dp[m] = max(dp[m], dp[m-v] + w, dp[m-2*v] + 2*w, dp[m-3*v] + 3*w, ...)
  接下来,我们把 dp[0] --> dp[m] 写成下面这种形式
  dp[0], dp[v],   dp[2*v],   dp[3*v],   ... , dp[k*v]
  dp[1], dp[v+1], dp[2*v+1], dp[3*v+1], ... , dp[k*v+1]
  dp[2], dp[v+2], dp[2*v+2], dp[3*v+2], ... , dp[k*v+2]
  ...
  dp[j], dp[v+j], dp[2*v+j], dp[3*v+j], ... , dp[k*v+j]

  因为k*v+j一定可以等于m,所以很容易想到可以把这些分为j个类,每一类中的值,都是在同类之间转换得到的,也就是说,dp[k*v+j] 只依赖于 { dp[j], dp[v+j], dp[2*v+j], dp[3*v+j], ... , dp[k*v+j] },因为我们需要的是{ dp[j], dp[v+j], dp[2*v+j], dp[3*v+j], ... , dp[k*v+j] } 中的最大值,可以通过维护一个单调队列来得到结果。这样的话,问题就变成了 j 个单调队列的问题。所以,我们可以得到:
  dp[j]    =     dp[j]
  dp[j+v]  = max(dp[j] +  w,  dp[j+v])
  dp[j+2v] = max(dp[j] + 2w,  dp[j+v] +  w, dp[j+2v])
  dp[j+3v] = max(dp[j] + 3w,  dp[j+v] + 2w, dp[j+2v] + w, dp[j+3v])
  ...
  但是,这个队列中前面的数,每次都会增加一个 w ,所以我们需要做一些转换
  dp[j]    =     dp[j]
  dp[j+v]  = max(dp[j], dp[j+v] - w) + w
  dp[j+2v] = max(dp[j], dp[j+v] - w, dp[j+2v] - 2w) + 2w
  dp[j+3v] = max(dp[j], dp[j+v] - w, dp[j+2v] - 2w, dp[j+3v] - 3w) + 3w
  ...
  这样,每次入队的值是 dp[j+k*v] - k*w

  key:

  维护队列元素的个数,如果不能继续入队,弹出队头元素
  维护队列的单调性,即:尾值 >= dp[j + k*v] - k*w

  代码:

#include <iostream>
#include <cstring>

using namespace std;

const int N = 20010;

int dp[N], pre[N], q[N];
int n, m;

int main() {
    cin >> n >> m;
    for (int i = 0; i < n; ++i) {
        memcpy(pre, dp, sizeof(dp));
        int v, w, s;
        cin >> v >> w >> s;
        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;

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

                if (head <= tail)
                    dp[k] = max(dp[k], pre[q[head]] + (k - q[head])/v * w);

                q[++tail] = k;
            }
        }
    }
    cout << dp[m] << endl;
    return 0;
}

 

posted on 2021-01-19 20:24  mibbp  阅读(167)  评论(0编辑  收藏  举报

导航