描述
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
思路
本题是0-1背包问题的一个变形。首先,若数组中的数总和sum为奇数,那么肯定不能分成两组相等的子集。想要分成两组相等的子集,那么每个子集的的和就是数组的总和的1/2,这样问题便转化为:有一个能装重量为sum/2的背包,现在有n件物品,每件物品的重量和价值相等,能否从中选出部分物品,每件物品只能装一次,恰好将背包装满?所以这道题不需要考虑价值,只要看是否能恰好装满即可。
还是0-1背包问题需要考虑的三件事:
状态有两个;在前i个物品中选择,能否刚好凑出重量j。dp[i][j]
表示在前i件物品中选择时,能否刚好凑出重量j。true为能,false为不能。面对第i件物品,有两种选择,将其加入背包或者不加入背包。同样分为当前背包容量j能否装下第i件物品的情况。若背包容量j不足以装下第i件物品,那么只能选择不装入。若容量j能装下第i件物品,则考虑要不要将其装入背包。若不把第i件物品加入背包,那么能否恰好装满容量为j的背包,取决于只考虑前i-1件物品时能否恰好装满容量为j的背包,若只考虑前i-1件就能恰好装满容量为j的背包,那么考虑前i件物品也能恰好装满容量为j的背包,否则亦然。若把第i件物品装入背包,那么便取决于在只考虑前i-1件物品时,容量为j - nums[i-1]
的背包能否被恰好装满。(nums索引从0开始,所以是i-1)若容量为j - nums[i-1]
的背包能恰好装满,那加上第i件物品的重量便恰好装满了,若容量为j - nums[i-1]
的背包也不能被恰好装满,那加上第i件物品也不能恰好装满容量为j的背包。
状态转移方程如下:dp[i][j] = dp[i-1][j] || dp[i-1][j - nums[i-1]]
。
解答
class Solution {
public:
bool canPartition(vector<int>& nums) {
int size = nums.size();
int sum = 0;
for(int num : nums) sum += num;
if (sum % 2 == 1) return false;
else sum /= 2;
vector<vector<bool> > dp(size + 1, vector<bool>(sum + 1, false));
for(int i = 0;i <= size; ++i) dp[i][0] = true;
for(int i = 1;i <= size; ++i)
for(int j = 1;j <= sum; ++j){
if(j >= nums[i-1]){//能装下
if(dp[i-1][j - nums[i-1]] == true) dp[i][j] = true;//放入背包
else dp[i][j] = dp[i-1][j]; //不放入背包
}
else dp[i][j] = dp[i-1][j];
}
return dp[size][sum];
}
};