AB40 【模板】01背包

描述

你有一个背包，最多能容纳的体积是V。

现在有n个物品，第i个物品的体积为v_ivi​ ,价值为w_iwi​。

（1）求这个背包至多能装多大价值的物品？

（2）若背包恰好装满，求至多能装多大价值的物品？

输入描述：

第一行两个整数n和V，表示物品个数和背包体积。

接下来n行，每行两个数v_ivi​和w_iwi​，表示第i个物品的体积和价值。

1 <= n,V,vi,wi <= 10001≤n,V,vi​,wi​≤1000

输出描述：

输出有两行，第一行输出第一问的答案，第二行输出第二问的答案，如果无解请输出0。

示例1

输入：

3 5
2 10
4 5
1 4

输出：14 9

说明：装第一个和第三个物品时总价值最大，但是装第二个和第三个物品可以使得背包恰好装满且总价值最大。

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

int main() {
    int n = 0, v = 0;
    cin >> n >> v;
    std::vector<int> vecv, vecw;
    int tempv, tempw, k = n;
    while (k--) {
        cin >> tempv >> tempw;
        vecv.push_back(tempv);
        vecw.push_back(tempw);
    }
    //第一问：
    vector<vector<int>> dp(n, vector<int>(v + 1, 0));
    for (int i = 0; i <= v; i++) {
        if (i >= vecv[i]) dp[0][i] = vecw[0];
    }
    for (int i = 1; i < n; i++) {
        for (int j = v; j >=vecv[i]; j--) {
            dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - vecv[i]] + vecw[i]);
        }
    }
    cout << dp[n - 1][v] << endl;
    //第二问：
    vector<vector<int>> dp2(n, vector<int>(v + 1, INT32_MIN));
    dp2[0][0] = 0;
    for (int i = 0; i <= v; i++) {
        if (i == vecv[i]) dp2[0][i] = vecw[0];
    }
    for (int i = 1; i < n; i++) {
        for (int j = 0; j <= v; j++) {
            //体积新增后要放满不能用则之前体积小时的有效值
            if (vecv[i] <= j) dp2[i][j] = max(dp2[i - 1][j],dp2[i - 1][j - vecv[i]] + vecw[i]);
        }
    }
    if (dp2[n - 1][v] > 0) cout << dp2[n - 1][v ] << endl;
    else cout << 0 << endl;
    return 0;
}
// 64 位输出请用 printf("%lld")