题目描述:
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
- 每个数组中的元素不会超过 100
- 数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
解题思路:
1. 递归,将数组从大到小排序(从小到大排序递归会超时),凑sum /2 的值。
代码如下:
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum = accumulate(nums.begin(),nums.end(),0);
if(sum % 2 != 0) return false;
int target = sum / 2;
sort(nums.begin(), nums.end(),greater<int>());
return dfs(nums, target, 0);
}
bool dfs(vector<int>& nums,int target, int start){
if(target == 0) return true;
if(nums[start] > target) return false;
for(int i = start; i < nums.size() && nums[i] <= target; ++i){
if(dfs(nums, target - nums[i], i + 1))
return true;
}
return false;
}
};
2. 动态规划, 设置一个vector<int> dp 来记录数组各元素相加的累计和,判断是否等于sum/2。
代码如下:
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum = accumulate(nums.begin(),nums.end(),0);
if(sum % 2 != 0) return false;
int target = sum / 2;
vector<int> dp(sum + 1, 0);
dp[0] = 1;
for(auto i : nums){
for(int j = sum; j >= 0; --j)
if(dp[j] == 1)
dp[i + j] = 1;
if(dp[sum / 2] == 1)
return true;
}
return false;
}
};