背包模版
代码实现
#include <iostream>
#include <vector>
using namespace std;
int knapsack(int capacity, vector<int>& weights, vector<int>& values) {
int n = weights.size();
// 定义并初始化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], values[i - 1] + dp[i - 1][j - weights[i - 1]]);
} else {
// 当前物品超过背包容量,不放入
dp[i][j] = dp[i - 1][j];
}
}
}
// 返回最优解
return dp[n][capacity];
}
int main() {
int capacity = 10;
vector<int> weights = {2, 3, 4, 5};
vector<int> values = {3, 4, 5, 6};
int result = knapsack(capacity, weights, values);
cout << "The maximum value is: " << result << endl;
return 0;
}
在这个示例中,我们使用一个整数`capacity`来表示背包的容量,使用两个向量`weights`和`values`分别表示物品的重量和价值。首先,我们定义并初始化一个大小为`n+1`x`capacity+1`的dp数组,用于存储子问题的最优解。然后,我们使用两层循环,外层循环遍历物品,内层循环遍历背包容量。根据当前物品的重量和价值,我们根据题目要求的选择策略,通过状态转移方程计算得到dp[i][j]的最优解。最后,我们返回`dp[n][capacity]`作为问题的最优解