编辑代码

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

int knapSack(vector<int> &weights, vector<int> &values, int n, int capacity) {
    // 初始化dp数组
    vector<vector<int> > dp(n + 1, vector<int>(capacity + 1, 0));

    // 遍历物品
    for (int i = 1; i <= n; i++) {
        // 遍历背包容量
        for (int j = 1; j <= capacity; j++) {
            if (weights[i - 1] <= j) {
                // 选择放或不放当前物品
                dp[i][j] = 
                    max(dp[i - 1][j], dp[i - 1][j - weights[i - 1]] + values[i - 1]);
            } else {
                // 当前物品放不进背包
                dp[i][j] = dp[i - 1][j];
            }
        }
    }
    // 返回结果
    return dp[n][capacity];
}

int main() {
    // 物品重量
    vector<int> weights = {2, 3, 4};
    // 物品价值
    vector<int> values = {3, 4, 5};
    // 背包大小
    int capacity = 5;
    // 物品数量
    int n = values.size();
    cout << knapSack(weights, values, n, capacity) << endl;
    return 0;
}