动态规划3-01背包问题(二维/一维)

//题目:背包问题
//
//给定一组物品,每个物品都有重量和价值,以及一个可以容纳的最大重量的背包。找出能够装入背包的最大价值。
//
//问题函数定义:
//
//cpp
//int knapsack(vector<int>& weights, vector<int>& values, int capacity);
//裁判测试程序样例:
//
//cpp
#include<iostream>
#include<stack>
#include <vector>
#include<string>
#include<algorithm>
#include<unordered_map>
#include <climits>
#include<queue>
#include<unordered_set>
#include<cctype>
using namespace std;
#define maxsize 100
int knapsack(vector<int>& weights, vector<int>& values, int capacity) {
    vector<vector<int>> dp(weights.size() + 1, vector<int>(capacity + 1, 0));
    for (int i = 1; i < weights.size() + 1; i++) {
        for (int j = 1; j < capacity + 1; j++) {
            if (weights[i - 1] > j) {
                dp[i][j] = dp[i - 1][j];
            }
            else {
                dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weights[i - 1]] + values[i - 1]);
            }
        }
    }
    return dp[weights.size()][capacity];
}
int main() {
    vector<int> weights = { 1, 2, 3 };
    vector<int> values = { 6, 10, 12 };
    int capacity = 5;
    int max_value = knapsack(weights, values, capacity);
    cout << "Maximum value that can be put in knapsack: " << max_value << endl;
    return 0;
}
//输入样例:
//
//1 2 3
//6 10 12
//5
//输出样例:
//
//Maximum value that can be put in knapsack : 22

//也可以使用一维解决

int knapsack(vector<int>& weights, vector<int>& values, int capacity) {
    vector<int> dp(capacity + 1, 0);
    for (int i = 1; i < weights.size() + 1; i++) {
        for (int j = capacity; j >= 1; j--) {
            if (weights[i - 1] <= j) {
                dp[j] = max(dp[j], dp[j - weights[i - 1]] + values[i - 1]);
            }
        }
    }
    return dp[capacity];
}
 

  • 6
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值