题目
给你一个 只包含正整数 的 非空 数组 nums 。
请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
分析
将此问题抽象成0-1背包问题,nums既是重量也是价值,先求出nums总和sum,sum如果为基数就绝对不可能满足要求,如果为偶数就开始动态规划,以下是动规五部曲:
1、dp[j]表示当背包容量为j时所装最大价值,本题相当于价值和重量相同
2、递推方程dp[j] = max(dp[j],dp[j-nums[i]]+nums[i],注意dp最大容量为(100*200)/ 2 + 1
3、初始化dp[0]即当背包容量为0时,所能装的最大价值为0
4、遍历顺序,用滚动数组,先物品,后背包,遍历背包时倒序遍历
5、打印dp数组,本题没有错,不打印
题解
解答一:
class Solution {
public:
bool canPartition(vector<int>& nums) {
vector<int> dp(10001,0);
int target = 0;
for(int i = 0;i < nums.size();++i){
target+=nums[i];
}
if(target%2==1)return false;
else target/=2;
dp[0] = 0;
for(int i = 0;i < nums.size();++i){
for(int j = target;j>=nums[i];--j){
dp[j] = max(dp[j],dp[j-nums[i]]+nums[i]);
}
}
if (dp[target] == target)return true;
else return false;
}
};
解答二:
#include<iostream>
using namespace std;
int nums[201];
int dp[20001];
bool canPartition(int n){
int target = 0;
for(int i = 0;i < n;++i){
target+=nums[i];
}
if(target%2==1)return false;
else target/=2;
dp[0] = 0;
for(int i = 0;i < n;++i){
for(int j = target;j>=nums[i];--j){
dp[j] = max(dp[j],dp[j-nums[i]]+nums[i]);
}
}
if (dp[target] == target)return true;
else return false;
}
int main(){
int n;
cin>>n;
for(int i = 0;i<n;++i){
cin>>nums[i];
}
cout<<canPartition(n)<<endl;
return 0;
}