给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
添加链接描述
可以用回溯暴力搜索出所有答案的,但最后超时了,也不想再优化了,放弃回溯,直接上01背包吧。背包的体积为sum / 2,背包要放入的商品(集合里的元素)重量为 元素的数值,价值也为元素的数值,背包如何正好装满,说明找到了总和为 sum / 2 的子集。背包中每一个元素是不可重复放入。
刚开始我将元素个数当作重量,数值当作价值,但是发现求价值的最大值并不能解决是否恰好等于sum/2的情况,dp[i]的数值一定是小于等于i的。
如果dp[i] == i 说明,集合中的子集总和正好可以凑成总和i,理解这一点很重要。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum=0,dp[20000],n=nums.size();
memset(dp,0,sizeof(dp));
for(int i=0;i<n;i++)
sum+=nums[i];
if(sum%2==1)
return false;
sum=sum/2;
for(int i=0;i<n;i++)
for(int j=sum;j>=nums[i];j--)
{
dp[j]=max(dp[j],dp[j-nums[i]]+nums[i]);
}
if(dp[sum]==sum)
return true;
else
return false;
}
};