DAY45 背包
0-1背包
题目描述
给你一个可装载重量为 W 的背包和 N 个物品,每个物品有重量和价值两个属性。其中第 i 个物品的重量为 wt[i],价值为 val[i],现在让你用这个背包装物品,最多能装的价值是多少?
思路
1.dp数组及含义
dp[i][w]:前i个物品放入背包,背包容量为w,最大价值为dp[i][w]
2.dp递推公式
物品的遍历 + 重量的遍历
物品依次遍历;
重量:
(1)第 i 个物品无法放入背包,dp[i][w] = dp[i - 1][w]
(2)第 i 个物品可以放入背包,选择
选择一: 不放入 dp[i][w] = dp[i - 1][w]
选择二: 放入 dp[i][w] = dp[i - 1][w - weight[i- 1]];//第 i 个物品的重量weight[i - 1]
3.dp初始化
dp初始化为全0
4.遍历方向
物品的遍历 + 重量的遍历
5.举例
代码
int knapsack(int W, int N, int wt[], int val[]) {
assert(N == sizeof(wt)/sizeof(wt[0])); // 确保N和wt的长度匹配
int dp[N + 1][W + 1]; // base case 已初始化为0
for (int i = 1; i <= N; i++) {
for (int w = 1; w <= W; w++) {
if (w - wt[i - 1] < 0) {
// 这种情况下只能选择不装入背包
dp[i][w] = dp[i - 1][w];
} else {
// 装入或者不装入背包,择优
dp[i][w] = std::max(
dp[i - 1][w - wt[i-1]] + val[i-1],
dp[i - 1][w]
);
}
}
}
return dp[N][W];
}
分割等和子集
题目描述
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
思路
1.dp数组及含义
dp[i][w]:前 i 个物品放入背包,背包容量为w,dp[i][w] = true,说明可以正好装满。
此处目标背包容量 w=sum/2,物品重量等于物品价值等于nums[i]。
2.dp递推公式
物品的遍历 + 重量的遍历
物品依次遍历;
重量:
(1)第 i 个物品无法放入背包,dp[i][w] = dp[i - 1][w]
(2)第 i 个物品可以放入背包,选择
选择一: 不放入 dp[i][w] = dp[i - 1][w]
选择二: 放入 dp[i][w] = dp[i - 1][w - weight[i- 1]];//第 i 个物品的重量weight[i - 1]
3.dp初始化
dp[i][0] 初始化为true,其他 false
4.遍历方向
物品的遍历 + 重量的遍历
5.举例
代码
class Solution {
public:
bool canPartition(vector<int>& nums) {
int n = nums.size();
int sum = 0;
for (int i = 0; i < n; i++)
{
sum += nums[i];
}
if (sum % 2) return false;
sum = sum / 2;
vector<vector<bool>> dp(n + 1, vector<bool>(sum + 1));
for (int i = 0; i <= n; i++)
{
dp[i][0] = true;
}
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= sum; j++)
{
//背包容量不足,不能装
if (j - nums[i - 1] < 0)
{
dp[i][j] = dp[i - 1][j];
}
else
{
//背包容量足够,选择装或不装
dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]];
}
}
}
return dp[n][sum];
}
};