题目:
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
思路:
可以用DFS+回溯的方法,但是时间复杂度较高,这里可以采用动态规划的方法,看做一个背包问题。dp[i][j]表示前i个物品是否可以填满容量为j的背包。状态转移方程如下,放入或者不放入第i个物品,以及背包容量不够的情况。
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= sum; j++) {
if (j - nums[i - 1] < 0) {
// 背包容量不足,不能装入第 i 个物品
dp[i][j] = dp[i - 1][j];
} else {
// 装入或不装入背包
dp[i][j] = dp[i - 1][j] || dp[i - 1][j-nums[i-1]];
}
}
}
代码:
class Solution {
public:
bool canPartition(vector<int>& nums) {
int len = nums.size();
if (len == 0) return false;
int sum = 0;
for (int n : nums)
sum += n;
if (sum % 2 == 1) return false;
vector<vector<bool>>dp(len + 1, vector<bool>(sum / 2 + 1,false));
for (int i = 0; i <= len; i++)
dp[i][0] = true;
for (int i = 1; i < len + 1; i++)
{
for (int j = 1; j <= sum / 2; j++)
{
if (j - nums[i - 1] < 0)
// 背包容量不足,不能装入第 i 个物品
dp[i][j] = dp[i - 1][j];
else
dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]];
}
}
return dp[len][sum/2];
}
};