编辑代码

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

void knapsack(vector<int>& value, vector<int>& weight, int n, int capacity) {
    // 创建二维数组dp并初始化为0
    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 (weight[i - 1] > j) {
                // 当前水果的重量大于背包容量,无法装入
                dp[i][j] = dp[i - 1][j];
            } else {
                // 装入或不装入当前水果
                dp[i][j] = max(dp[i - 1][j - weight[i - 1]] + value[i - 1],
                               dp[i - 1][j]);
            }
        }
    }

    // 输出装水果的策略
    cout << "装入水果的策略:" << endl;
    int i = n, j = capacity;
    while (i > 0 && j > 0) {
        if (dp[i][j] != dp[i - 1][j]) {
            // 装入了当前水果
            cout << "装入第" << i << "个水果" << endl;
            j -= weight[i - 1];  // 更新背包容量
        }
        i--;
    }
    // 输出装水果的总价值
    cout << "总价值:" << dp[n][capacity] << endl;
}

int main() {
    // 水果数量
    int n = 4;
    // 背包最大容量
    int capacity = 20;
    // 创建水果的重量和价值数组
    vector<int> weight = {15, 18, 10, 9};
    vector<int> value = {300, 180, 150, 270};
    // 求解装水果策略
    knapsack(value, weight, n, capacity);
    return 0;
}