给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
方法一 深搜(超时)
class Solution {
public:
int sum, len;
bool flag;
vector<vector >dp(201, vector())
void f(vector<int>& nums, int p, int s)
{
if (p >= len || s < 0) return;
if (s == 0){
flag = true;
return;
}
for (int i = p; i < len; ++i)
{
if (nums[i] <= s)
{
f(nums, p + 1, s);
f(nums, p + 1, s - nums[i]);
}
}
}
bool canPartition(vector<int>& nums) {
int i;
len = nums.size();
sum = accumulate(nums.begin(), nums.end(), 0);
if (sum % 2) return false;
sum /= 2;
f(nums, 0, sum);
return flag;
}
};
方法二:动态规划
class Solution {
public:
bool canPartition(vector<int>& nums) {
int i, j, len, sum;
len = nums.size();
sum = accumulate(nums.begin(), nums.end(), 0);
vector<bool>dp(sum, false);
if (sum % 2) return false;
sum /= 2;
dp[0] = true;
for (i = 0; i < len; ++i)
{
for (j = sum; j >= nums[i]; --j)
{
dp[j] = dp[j] || dp[j-nums[i]];
}
}
return dp[sum];
}
};