01背包问题
01背包问题,也就是说,可以向限定容量的背包当中放入物体,每个物体只有放和不放两种状态,每个物体都有自己的重量以及价值。现在将物体放入背包当中,要求重量总和不超过背包的容量,而且要求总价值最高。
这个问题可以由二维数组和一维数组的方式来解决,同样是五步。首先是二维数组:
1、dp[i][j]代表从编号0-i的物体当中选择任意个物体,放入背包容量为j的背包里,价值最大为多少。
2、dp[i][j] = max(dp[i-1][j], dp[i-1][j-weight[i]] + value[i]) 很好理解,就是加和不加的两种情况。
3、初始化,第一列全为0,第一行当背包容量大于等于第一个物体的重量的时候,为第一个物体的价值,其余为0。其他所有位置初始化为0即可。
4、前向遍历,外层嵌套物体id,内层嵌套背包重量,这样就可以不断的加物体。
5、模拟.
其实一维数组和二维数组是类似的,就是遍历顺序是从后向前遍历,因为从前向后遍历的过程中,会有一些参考值被覆盖掉,导致误加的情况发生。
416. 分割等和子集
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意: 每个数组中的元素不会超过 100 数组的大小不会超过 200
示例 1: 输入: [1, 5, 11, 5] 输出: true 解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2: 输入: [1, 2, 3, 5] 输出: false 解释: 数组不能分割成两个元素和相等的子集.
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
这道题其实就是要找一个背包,这个背包可以放下一半重量的物体。这里物体的重量和价值是一样的,这里把背包极限设为物体总重量的一半。所以当物体总重量达到背包极限的时候,其价值就是物体重量可以到达的极限。当两个极限相同时,说明可以分成两半。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int target = accumulate(nums.begin(), nums.end(), 0);
if(target % 2 == 1) return false;
vector<int> dp(target/2+1);
for(int i = 0; i < nums.size(); i++){
for(int j = target/2; j >= nums[i]; j--){
dp[j] = max(dp[j], dp[j-nums[i]] + nums[i]);
}
}
if(dp[target/2] == target/2) return true;
return false;
}
};
1049. 最后一块石头的重量 II
和上一道题一致,直接给出代码即可。
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
int target = accumulate(stones.begin(), stones.end(), 0);
vector<int> dp(target/2+1);
for(int i = 0; i < stones.size(); i++)
for(int j = target/2; j >= stones[i]; j--)
dp[j] = max(dp[j], dp[j-stones[i]]+stones[i]);
return target-dp[target/2]*2;
}
};