给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
示例 2:
输入: [1, 2, 3, 5]
输出: false
解释: 数组不能分割成两个元素和相等的子集.
思路:
为什么说这道题可以转为背包问题?我觉得可以这样理解:
(1)已知整个序列,那么我们可以由计算出每个集合的元素和是多少;=>背包的总容量cap
(2)要对每一个元素进行选择,决定元素要放在两个集合中的哪一个;=>物品的两种状态:放进背包与不放进背包
所以问题转化为将能否将n个物品中的部分物品放进背包中使得背包总容量为cap。
设定数组bag[w]表示容量为w的背包可以放进价值为多少的物品,然后将序列中的每个元素ele视为重量和价值都相同的物品。
代码:
#include<iostream>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
bool canPartition(vector<int>& nums) {
int nums_len=nums.size(),nums_sum=0;
for(int i=0;i<nums_len;++i){nums_sum+=nums[i];}
if(nums_sum&1)return false;
nums_sum=(nums_sum>>1);
vector<int>dp(nums_sum+1,0);
for(int w:nums){
for(int v=nums_sum;v>=w;--v){
if(dp[v]<dp[v-w]+w){
dp[v]=dp[v-w]+w;
}
}
}
return dp[nums_sum]==nums_sum;
}
};