01背包问题二维
视频讲解链接:https://www.bilibili.com/video/BV1cg411g7Y6/
01背包问题一维
视频讲解链接:https://www.bilibili.com/video/BV1BU4y177kY/?spm_id_from=333.788
力扣416.分割等和子集
题目链接:https://leetcode.cn/problems/partition-equal-subset-sum/
思路
如果能分成两个子集,且子集内和相等,那么可以定义target等于nums数组和的一半,只要子集里的和等于target,返回true.
1.确定dp[j]的含义:背包容量为j的背包最大价值为dp[j]
2.递归过程:
(1)不放物品i,那么背包容量还是j,它所背的最大价值还是dp[j];
(2)放物品i,那么放进物品i之前的背包所背的最大价值是dp[j-weight[i]],再加上物品i的价值,就是现在背包所背的最大价值,即dp[j-weight[i]] + value[i]。
在本题中,物品的重量和价值是相等的,都是nums[i]。所以为:dp[j-nums[i]]+nums[i]。每次应该取和本身比较的最大值,即dp[j] = Math.max(dp[j],dp[j - nums[i]]+nums[i])
3.初始化:dp[0] = 0,背包容量为0,所装下的最大价值dp[0]也为0.其余值初始化为0,因为在遍历中,会取最大值,如果初始化值太大会覆盖原本的结果。
4.遍历顺序:先遍历物品,再遍历背包。
为什么for (int j = target; j >= nums[i] ; j--)
,遍历背包是倒序的?
参考01背包问题一维的视频讲解,因为如果正序,背包会重复放进一个物品,模拟一下就可以知道,而每个物品只有一个。判断条件是j >= nums[i]
,因为当背包容量小于要放进的物品重量时,物品i就放不进背包了。
5.打印数组:当最后target值等于背包在容量为target时的最大价值时,即target == dp[target]
,返回true。
完整代码
class Solution {
public boolean canPartition(int[] nums) {
int sum = 0;
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
}
if(sum % 2 == 1) return false;
int target = sum / 2;
int[] dp = new int[target+1];
dp[0] = 0;
for (int i = 0; i < nums.length; i++) {
for (int j = target; j >= nums[i] ; j--) {
dp[j] = Math.max(dp[j],dp[j - nums[i]]+nums[i]);
}
}
if(target == dp[target]){
return true;
}else{
return false;
}
}
}