这道题按如下推导,把它转化成求是否存在子集的和为特定值。跟416. Partition Equal Subset Sum一样的做法。这里给出一维数组的简化版。
这题的坑在于,要判断数组的和是否大于s,如果小于s,直接返回0,样例中有s=1000000000的情况,很坑爹
我们是在一些数前面添加+,一些数前面添加-。我们把添加+的数记为集合P,添加-的数记为集合N。
比如{1, 2, 3, 4, 5}, target = 3。如果我们这么添加:+1 - 2 + 3 - 4 + 5 = 3。则P = {1, 3, 5}, N = {2, 4}。
那么,我们可以得到如下公式:
(1)sum(P)−sum(N)=target
(2)sum(P)+sum(N)=sum
(1)+(2)
得到:
2sum(P)=target+sum
所以,我们的问题就转化成了:在nums[]
找一个子集P,使P的和为
targt+sum2
。
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int S) {
int target = 0;
for(int i = 0; i < nums.size(); i++)
target += nums[i];
if((target + S)%2 == 1 || target < S)return 0;
target = (target + S) >> 1;
int d[target + 1] = {0};
d[0] = 1;
for (int i = 0; i < nums.size(); ++i)
{
for (int j = target; j >= nums[i]; --j){
d[j] += d[j - nums[i]];
}
}
return d[target];
}
};