刷题模板-背包问题

一、0/1背包问题

二维dp数组

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

const int N = 1010;
// n表示物品数目  m表示总体积大小
int n, m;
int dp[N][N];
int v[N], w[N];

int main(){
    cin >> n >> m;
    for(int i=1; i<=n; i++) cin >> v[i] >> w[i];
    // 由于dp是全局变量 所以初始化都是0 不需要手动初始化
    for(int i=1; i<=n; i++){
        int vol = v[i];
        for(int j=1; j<=m; j++){
            dp[i][j] = dp[i-1][j];
            if(j >= vol) dp[i][j] = max(dp[i][j], dp[i-1][j - vol] + w[i]);
        }// end for
    }// end for

    cout<< dp[n][m] <<endl;
    return 0;
}

降维优化, 降为1维dp数组

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

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


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

二、完全背包问题

每一个物品都可以选择多次（当然最终的体积还是不能超过背包的最大体积）

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

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

int main(){
    cin >> n >> m;
    for(int i=1; i<=n; i++) cin>> v[i] >> w[i];

    /*
        状态转移方程：
        将集合划分为：第i个物品选0次  第i个物品选1次  ... 第i个物品选k-1次  第i个物品选k次
        dp[i][j]=max(dp[i-1][j], dp[i-1][j - v[i]] + w[i],...,dp[i-1][j-k*v[i]]+k*w[i]);
    */
    for(int i=1; i<=n; i++){
        int vol = v[i];
        for(int j=1; j<=m; j++){
            for(int k = 0; k*vol <= j; k++){
                dp[i][j] = max(dp[i][j], dp[i-1][j - k*vol] + k*w[i]);
            }// end for
        }// end for
    }// end for    

    cout << dp[n][m] <<endl;
    return 0;
}

完全背包问题进行优化， 将dp降为二维。

dp[i][j] =    max(dp[i-1][j], dp[i-1][j-v[i]]+w[i], ..., dp[i-1][j-k*v[i]]+k*w[i])
dp[i][j-v[i]]=max(            dp[i-1][j-v[i]], ...,      dp[i-1][j-k*v[i]]+(k-1)*w[i])
因此：
dp[i][j] = max(dp[i-1][j], dp[i][j-v[i]]+w[i])

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

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

int main(){
    cin>> n >> m;
    for(int i=1; i<=n; i++) cin >> v[i] >> w[i];

    /*
        状态转移方程：
            dp[i][j] = max(dp[i-1][j], dp[i][j - v[i]] + w[i]);
    */
    for(int i=1; i<=n; i++){
        int vol = v[i];
        for(int j = 1; j<=m; j++){
            dp[i][j] = dp[i-1][j];
            if(j >= vol) dp[i][j] = max(dp[i][j], dp[i][j-vol] + w[i]);
        }// end for
    }//end for    
    
    cout << dp[n][m] <<endl;
    return 0;

}

完全背包继续优化，将dp将为1维

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

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

int main(){
    
    cin >> n >> m;
    for(int i=1; i<=n; i++) cin >> v[i] >> w[i];
    
    /*
        状态转移方程
             dp[i][j] = max(dp[i-1][j], dp[i][j-v[i]]+w[i]);
                                 ||
                                 ||
             dp[j] = max(dp[j], dp[j-v[i]]+w[i])
             dp[j-v[i]] 会在 dp[j]之前计算好 并被赋值
    */
    for(int i=1; i<=n; i++){
        int vol = v[i];
        for(int j=vol; j<=m; j++){
            dp[j] = max(dp[j], dp[j-vol] + w[i]);  
        }// end for
    }// end for

    cout << dp[m] <<endl;
    return 0;
}