Leetcode C++《热题 Hot 100-59》416. 分割等和子集
1. 题目
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-equal-subset-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 思路
- 总和等于sum,选择一些数的和使得它等于sum/2
- dp[i][j] 关键是状态转移方程
- 不选这个数 dp[i][j] = 1 if dp[i-1][j]=1
- 选择这个数 dp[i][j] = 1 if dp[i-1][j-nums[i]] = 1
- 时间复杂度kn,k为数字之和
3. 代码
class Solution {
public:
bool canPartition(vector<int>& nums) {
//最多200*100
bool **dp = new bool*[200];
int sum = 0;
for (int i = 0; i < nums.size(); i++) {
sum += nums[i];
}
if (sum%2 == 1)
return false;
for (int i = 0; i < 200; i++)
dp[i] = new bool[sum+1];
dp[0][0] = true;
for (int i = 0; i < nums.size(); i++) {
for (int j = 0; j < sum+1; j++) {
if (i >0 && dp[i-1][j]) {
dp[i][j] = true;
continue;
}
if (i >0 && j-nums[i] >= 0) {
if (dp[i-1][j-nums[i]]) {
dp[i][j] = true;
continue;
}
}
if (i == 0 && j == nums[i])
dp[i][j] = true;
else
dp[i][j] = false;
}
}
for (int i = 0; i < nums.size(); i++) {
if (dp[i][sum/2])
return true;
}
return false;
}
};