给你一个 只包含正整数 的 非空 数组 nums
。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5] 输出:true 解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5] 输出:false 解释:数组不能分割成两个元素和相等的子集。
个人分析:
这个题是01背包的应用,将一个子集的和相加得到sum,再将sum/2得到target,数组中的数字就相当于价值和重量相同的物品,而target就是背包的容量,我们就需要在数组中所有的数字中挑选出合适的数字来装满容量为target的背包,而与01背包不同的是它的重量和价值相等,所以挑选的时候就是挑选的物品重量尽可能的贴近书包的容量。最后如果可以得到恰好完全装满的背包,就说明该数组可以被分割成两个完全相等的子集。
题解代码:
class Solution {
public:
bool canPartition(vector<int>& nums) {
int n = nums.size();
if(n < 2)
{
return false;
}
int sum = 0;
int Max = 0;
for(int i = 0; i < n; i++)
{
sum = sum + nums[i];
if(nums[i] > Max)
{
Max = nums[i];
}
}
if(sum % 2 != 0)
{
return false;
}
int target = sum / 2;
if(Max > target)
{
return false;
}
int dp[n + 1][target + 1];
for(int i = 0; i < n + 1; i++)
{
for(int j = 0; j < target + 1; j++)
{
dp[i][j] = 0;
}
}
for(int i = 1; i <= n; i++)//物品
{
for(int j = 1; j <= target; j++)//背包
{
if(j < nums[i - 1])
{
dp[i][j] = dp[i - 1][j];
}
else
{
dp[i][j] = max(dp[i -1][j], dp[i -1][j - nums[i - 1]] + nums[i - 1]);
}
}
}
if(dp[n][target] == target)
{
return true;
}
else
{
return false;
}
}
};