动态规划,一维版本
dp[i][j]
表示是否存在[0,i]中取一些元素使其和为j
则dp[i][j] = dp[i-1][j] || dp[i-1][s-nums[i]]
(分别表示取i与不取i的情况)
然后第i行的值只依赖于第i-1行,因此可以轻易缩减为2行,
如果从右向左求值,则只需1行。
bool canPartition(vector<int>& nums) {
if(nums.size()==0||nums.size()==1)
return false;
int sum=accumulate(nums.begin(),nums.end(),0);
if(sum%2) return false;
int n = nums.size();
int target = sum/2;
vector<bool> dp(target+1,false);
dp[0] = true;
for(int i=0;i<n;++i){
for(int j=target;j>=nums[i];--j){
dp[j] = dp[j] || dp[j-nums[i]];
}
}
return dp[target];
}
剪枝,更快
class Solution {
public:
int target;
bool check(vector<int> &nums, int cur1,int cur2,int index){
//剪枝,如果有一个和超过了target,则停止搜索(这个剪枝非常重要)
if(target < cur1 || target < cur2 || index == -1) {
return false;
}
if(target == cur1||target == cur2) {
return true;
}
//将nums[index]分配cur1或者cur2
return check(nums, cur1 + nums[index], cur2, index-1)
|| check(nums, cur1, cur2 + nums[index], index-1);
}
bool canPartition(vector<int>& nums) {
//accumulate带有三个形参:头两个形参指定要累加的元素范围,第三个形参则是累加的初值。
int sum=accumulate(nums.begin(),nums.end(),0);
if(sum % 2 != 0){
return false;
}
target = sum / 2;
return check(nums, 0, 0, nums.size() - 1);
}
};