完全背包问题浅谈

题面

样例

朴素版代码

思路

跟01背包相似，我们把f[i][j]分为k类，分别是含0个第i个物品，含1个，含2个.......含k个。那么f[i][j]的更新方程就是f[i][j] = f[i-1][j-v[i]*k]+w[i]*k(针对于第k类而言，同样要有j>=k*v[i]的条件限制)

代码

#include<iostream>

using namespace std;

const int N = 1010;
int v[N],w[N];
int n,m;
int f[N][N];

int main(void){
    cin >> n >> m;
    
    for(int i=1;i<=n;i++) cin >> v[i] >> w[i];
    
    for(int i=1;i<=n;i++){
        for(int j=0;j<=m;j++){
            for(int k=0;k*v[i] <= j;k++){
                f[i][j] = max(f[i][j],f[i-1][j-v[i]*k]+w[i]*k);
            }
        }
    }
    
    cout << f[n][m];
    
    return 0;
}

半优化代码

思路

我们把f[i][j]的更新公式展开，它含有k+1项，且后k项形式一样，那么我们不妨再展开f[i][j-v]惊奇地发现它刚好与f[i][j]的后k项相似(每项只差一个w)，那么也就意味着后k项即为f[i][j-v]再加上w即可。于是即可省去k循环，此时代码于01背包问题的朴素代码一模一样。

代码

#include<iostream>

using namespace std;

const int N = 1010;
int v[N],w[N];
int n,m;
int f[N][N];

int main(void){
    cin >> n >> m;
    
    for(int i=1;i<=n;i++) cin >> v[i] >> w[i];
    
    for(int i=1;i<=n;i++){
        for(int j=0;j<=m;j++){
            f[i][j] = f[i-1][j];
            if(j>=v[i]) f[i][j] = max(f[i][j],f[i][j-v[i]]+w[i]);
        }
    }
    
    cout << f[n][m];
    
    return 0;
}

优化代码

思路

同01背包问题一样，此时我们可以优化二层循环内部代码，具体见01背包问腿那篇博文，但此时更新的顺序要从小到大了(01背包问题中是从大到小)，因为完全背包问题每次都需要第k次的数据(01背包·用的是k-1次)，所以直接从小到大即可。

代码

#include<iostream>

using namespace std;

const int N = 1010;
int v[N],w[N];
int n,m;
int f[N][N];

int main(void){
    cin >> n >> m;
    
    for(int i=1;i<=n;i++) cin >> v[i] >> w[i];
    
    for(int i=1;i<=n;i++){
        for(int j=v[i];j<=m;j++){
            f[j] = max(f[j],f[j-v[i]]+w[i]);
        }
    }
    
    cout << f[n][m];
    
    return 0;
}