【问题描述】
Given a non-empty array containing only positive integers, find if the array can be partitioned into two subsets such that the sum of elements in both subsets is equal.
Note:
- Each of the array element will not exceed 100.
- The array size will not exceed 200.
Example 1:
Input: [1, 5, 11, 5] Output: true Explanation: The array can be partitioned as [1, 5, 5] and [11].
Example 2:
Input: [1, 2, 3, 5] Output: false Explanation: The array cannot be partitioned into equal sum subsets.问题来源:partition-equal-subset-sum
【解题分析】
题意是让我们把一个数组划分成两个子数组,要求两个子数组的和相等。
假设原来数组的和为sum,数组大小为n。那么我们不妨这样假设,从原来的数组中挑选i个数(i≤n),使得它们的和为sum/2,这样便成了很经典的0/1背包问题。
状态表示:dp[i][j]代表前i个数中和为j的方案是否可行,0代表不可行,1代表可行。
状态转移方程:dp[i][j] |= dp[i-1][j] + dp[i-1][j-nums[i]],dp[0][0] = 1
返回结果:dp[n][sum/2]
我们再对它进行优化:
我们对它进行降维处理,从dp[i][j]降成dp[j],循环n次去更新dp[j]。
状态转移方程:dp[j] |= dp[j-nums[i]],i是当前循环的次数,i从0到n-1。
返回结果:dp[sum/2]
时间复杂度O(N^2),空间复杂度O(N)。
【源代码】
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum = 0;
for (auto n : nums) {
sum += n;
}
if (sum % 2 != 0) {
return false;
}
int target = sum / 2;
vector<int> dp(target + 1, 0);
dp[0] = 1;
for (int i = 0; i < nums.size(); i++) {
for (int j = target; j >= nums[i]; j--) {
dp[j] |= dp[j - nums[i]];
}
}
if (dp[target] != 0) {
return true;
}
return false;
}
};