代码随想录算法训练营33期 第四十二天 |01背包问题 二维、01背包问题 一维

01背包问题 二维

//二维dp数组实现
#include <bits/stdc++.h>
using namespace std;

int n, bagweight;// bagweight代表行李箱空间
void solve() {
    vector<int> weight(n, 0); // 存储每件物品所占空间
    vector<int> value(n, 0);  // 存储每件物品价值
    for(int i = 0; i < n; ++i) { //i是物品种类
        cin >> weight[i];
    }
    for(int j = 0; j < n; ++j) { //j是总容量
        cin >> value[j];
    }
    vector<vector<int>> dp(n, vector<int>(bagweight+1));
    
    //初始化
    for (int i=0; i<n; i++){ //列
        dp[i][0]=0;
    }
    for (int j=0; j<bagweight+1; j++){ //行
        if (j>=weight[0]){
            dp[0][j]=value[0];
        }
    }
    for (int i=0; i<n; i++){
        for (int j=0; j<bagweight+1; j++){
            cout << dp[i][j] <<" ";
        }
        cout << endl;
    }
    cout << endl;// done
    
    for (int i=1; i<n; i++){
        for (int j=1; j<bagweight+1; j++){
            if (j-weight[i]<0) dp[i][j] = dp[i-1][j]; //背包无法装得下这个物品，那么就继承没装物体时的总价值
            else dp[i][j] = max(dp[i-1][j], dp[i-1][j-weight[i]]+value[i]); // dp[i-1][j-weight[i]没装这个物品时的最高价格（因为是最左侧）
        }
    }
    
    for (int i=0; i<n; i++){
        for (int j=0; j<bagweight+1; j++){
            cout << dp[i][j] <<" ";
        }
        cout << endl;
    }
    cout << endl;
    
    cout<<dp[n-1][bagweight];
}

int main() {
    while(cin >> n >> bagweight) {
        solve();
    }
    return 0;
}

46. 携带研究材料

// 一维dp数组实现
#include <iostream>
#include <vector>
using namespace std;

int main() {
    // 读取 M 和 N
    int M, N; //M是材料种类，N是背包空间
    cin >> M >> N;

    vector<int> costs(M);
    vector<int> values(M);

    for (int i = 0; i < M; i++) {
        cin >> costs[i];
    }
    for (int j = 0; j < M; j++) {
        cin >> values[j];
    }

    // 创建一个动态规划数组dp，初始值为0
    vector<int> dp(N + 1, 0);
    
    for (int i=0; i<M; i++){ //遍历每种物体
        for (int j=N; j>=costs[i]; j--){ //对每种物品都尝试（j>=costs[i]）放入背包中，能成功放入就更新dp
            dp[j] = max(dp[j-1], dp[j-costs[i]]+values[i]);
            cout << "dp[" << j <<"] = " << dp[j] << " ";
        }
    }
    cout << endl;
    cout << endl;
    for (int i=0; i<dp.size(); i++){
        cout << dp[i] << " ";
    }
    cout << endl;

    // 外层循环遍历每个类型的研究材料
    cout << dp[N];

    return 0;
}