416. 分割等和子集
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
从普通问题中抽象出01背包
状态压缩型二维到一维的写法
重量就是价值,价值就是重量的背包问题
示例 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
class Solution {
public:
bool canPartition(vector<int>& nums) {
// dp[i]中的i表示背包内总和
// 题目中说:每个数组中的元素不会超过 100,数组的大小不会超过 200
// 总和不会大于20000,背包最大只需要其中一半,所以10001大小就可以了
vector<int> dp(10001, 0);
// int sum = 0;
// for (int i = 0; i < nums.size(); i++)sum += nums[i];
// 也可以使用库函数一步求和
int sum = accumulate(nums.begin(), nums.end(), 0);
if (sum % 2 == 1) return false;//剪枝:不能切成两个和相等的子集
int target = sum / 2;
// 开始 01背包
for(int i = 0; i < nums.size(); i++) {//物品
//从满足的容量逆推到至少能装下该物品的容量
for(int j = target; j >= nums[i]; j--) { //逆序背包:状态压缩,二维dp->一维dp时必须逆序背包,保证每一个元素一定是不可重复放入
dp[j] = max(dp[j], dp[j - nums[i]] + nums[i]);
if(i==nums.size()-1)break;//剪枝:只需要算出for遍历到最后一层物品时的dp[target]就行
}
}
// 集合中的元素正好可以凑成总和target
if (dp[target] == target) return true;
return false;
}
};