[代码随想录] 01背包问题

零一背包问题
https://kamacoder.com/problempage.php?pid=1046

//竞赛中常用，
//可以直接使用STL（标准模板库）中的数据结构和算法
#include <bits/stdc++.h>
using namespace std;

int n, bagweight;

void solve(){
    // 先初始化变量
    vector<int> weight(n, 0);
    vector<int> value(n, 0);
    
    for (int i = 0; i < n; i++) {
        cin >> weight[i];
    }
    for (int j = 0; j < n; j++) {
        cin >> value[j];
    }
    
    // 初始化dp, 背包容量是0到bagweight 所以要 + 1
    vector<vector<int>> dp(n, vector<int> (bagweight + 1, 0));
    for (int i = weight[0] ; i <= bagweight; i++) {
        dp[0][i] = value[0];
    }
    // 其实也应该初始化第一列，但是前面已经做过了这一步！
    // 注意这里的下标，n个物品最大下标为n - 1
    for (int i = 1; i < n ;i++) {
        // 背包的容量取值是0到bagweight
        for ( int j = 1; j <= bagweight; j++) {
            //分为放得下和放不下2种情况
            // 当前容量整个背包都放不下，那么只能继承上一层的最优放法
            if (weight[i] > j) {
                dp[i][j] = dp[i - 1][j];
            }else{
                // 如果放的下，又分为放还是不放两种情况取最大值
                dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
            }
        }
    }
    cout << dp[weight.size() - 1][bagweight] << endl;
}
int main()  {
    while (cin >>n >> bagweight) {
        solve();
    }
    return 0;
}