【c/c++算法刷题笔记】—— 背包问题2

【前言】OJ：AcWing

01背包问题

二维模板

若求恰好等于 m 的最大价值， f[0][0]=0,其余为INT_MIN

代码

#include <iostream>
#include <algorithm>
using namespace std;

const int N=1010;
int f[N][N];
int v[N],w[N];
int n,m;
int main(){
    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-1][j-v[i]]+w[i]);
        }
    }
    // int res=0;
    // for(int i=0;i<m;i++) res=max(res,f[n][i]);
    cout<<f[n][m]<<endl;
    return 0;
}

【优化】一维模板

（想不通就多想几遍）

Q1：第二层 容量j 为什么要从大到小循环？

A：从大到小循环的原因是 f [i,j] 要用 f [i - 1, j - v[i]] + w [i] 来更新，即，要利用前面一轮的结果。

下标 j 大于 j-v[i] ，从大到小循环时，先算 f [j]，再算 f [j - v[i]] ，可以保证算 f [j] 时，这一轮的 f [j - v[i]] 还没有更新，所以用到的 f [j - v[i]] 存储的是上一轮的值 —— f [i - 1][ j - v[i]] ；
但，如果从小到大循环，下标 j 大于 j-v[i] ，f [j - v[i]] 会在 f [j] 前被计算出来，那么它就表示 当前新一轮的值 —— f [i][ j - v[i]] 了。优化之后 f [N] 就只有一维了，下标 j 为容积， j - v[i] 比 j 小， f [j - v[i]] 会在 f [j] 之前被算出来。

代码

#include <iostream>
#include <algorithm>
using namespace std;

const int N=1010;
int f[N];
int v[N],w[N];
int n,m;
int main(){
    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=m;j>=v[i];j--){
            f[j]=max(f[j],f[j-v[i]]+w[i]);
        }
    }
    cout<<f[m]<<endl;
    return 0;
}

完全背包

笔记

每件物品可以选无限次

f[i] ：总体积是 i 的情况下，最大价值是多少

res = max(f[0...m])

转移方程
for(int i=0;i<n;i++){
    for(int j=v[i];j<=m;j++) f[j]=max(f[j],f[j-v[i]]+w[i]);
    
    for(int j=m;j>=v[i];j--){
        for(int k=0;k*v[i]<=j;k++){
            f[j]=max(f[j],f[j-k*v[i]]+k*w[i]);
        }
    }
}

数学归纳法：
1 假设考虑前 i-1 个物品之后，所有的f[j]都是正确的
2 来证明：考虑完第 i 个物品之后，所有的f[j]也都是正确的

对于某个 j 而言，如果最优解中包含 k 个v[i]

f[j-k*v[i]];

f[j-(k-1)*v[i]-v[i]]+w[i] //包含 1 个v[i]

...

f[j]f[j-v[i]]+w[i]

代码

#include <iostream>
#include <algorithm>
using namespace std;

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

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

多重背包

笔记

多重背包可看作**01背包**的扩展。

f[i] 总体积是 i 的情况下 ，最大价值是多少

for(int i=0;i<n;i++) //枚举物品
{
    for(int j=m;j>=v[i];j--){ //枚举背包容量
        f[j]=max( f[j], f[j-v[i]] + w[i], f [j-f [j- 2*v[i]] + 2*w[i],...,... );
    }
}
	
数组初始化不同，得到结果的方式也不同：
 f [i]=0;数组全初始化为0， f [m] 即为结果
 f [0]=0; 其余位置 -∞，结果为 max( f[0~m] )

代码

#include <iostream>
using namespace std;

const int N=1010;
int f[N];
int n,m;
int main(){
    cin>>n>>m;
    for(int i=0;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;
}