416 分割等和子集
题目链接:416.分割等和子集
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
思路:
本题很容易想到,先获取数组总sum,然后sum/2,从原数组中选数,使得总和恰好为sum/2。因此,原问题就变为01背包问题。
但使得总和恰好为sum/2,有点不清楚dp数组怎么转移了,其实本题中总和既是weight,也是value,因此,判断dp[sum/2]是否等于sum/2即可。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int len = nums.size();
int sum = 0;
for(int i = 0; i < len; i++)
{
sum += nums[i];
}
if(sum % 2 != 0) return false;
int target = sum / 2;
vector<int> dp(10001, 0);
for(int i = 0; i < nums.size(); i++) {
for(int j = target; j >= nums[i]; j--) { // 每一个元素一定是不可重复放入,所以从大到小遍历
dp[j] = max(dp[j], dp[j - nums[i]] + nums[i]);
}
}
if (dp[target] == target) return true;
return false;
}
};
本题使用了一维dp,建议先学习代码随想录中0/1背包的视频,才更容易理解。