给你一个非负整数数组 nums
和一个整数 target
。
向数组中的每个整数前添加 '+'
或 '-'
,然后串联起所有整数,可以构造一个 表达式 :
- 例如,
nums = [2, 1]
,可以在2
之前添加'+'
,在1
之前添加'-'
,然后串联起来得到表达式"+2-1"
。
返回可以通过上述方法构造的、运算结果等于 target
的不同 表达式 的数目。
示例 1:
输入:nums = [1,1,1,1,1], target = 3 输出:5 解释:一共有 5 种方法让最终目标和为 3 。 -1 + 1 + 1 + 1 + 1 = 3 +1 - 1 + 1 + 1 + 1 = 3 +1 + 1 - 1 + 1 + 1 = 3 +1 + 1 + 1 - 1 + 1 = 3 +1 + 1 + 1 + 1 - 1 = 3
示例 2:
输入:nums = [1], target = 1 输出:1
思路:这道题第一眼看到可能会觉得是回溯暴力解决,但各种排列组合的种类实在太多,肯定会超时。
我们可以想:给我们一个数组,若数组中全部数求和都小于target,那肯定没有方法,返回0;然后就是正常的情况了,对数组中部分的数取正,另外的数字取负,我们把取正的部分和记为x,因此取负的那部分和为sum-x,然后要使得x -(sum-x ) = target,我们可以移项,得到2x = (sum + target),因此sum+target一定得是个偶数,不然也返回0。至此,我们就把这道很复杂的题目转换成从数组中取几个数,让它们求和等于(sum + target)/2,就不用去管数组中每个数到底是取正还是取负了。
因为题目问我们是有几种方法,在之前做的0-1背包都是得到有没有一种方法能实现,那这里要求所有的方法,因此dp[j]要把之前所有的方法数量累加起来。
代码(Python):
class Solution(object):
def findTargetSumWays(self, nums, target):
sumNum = sum(nums)
if (sumNum + target) % 2 == 1: #若target+sumNum为奇数,找不出x,返回0
return 0
if target > sumNum: #如果数组总和小于target,返回0
return 0
x = (sumNum + target) / 2
dp = [0] * (sumNum +1)
dp[0] = 1
for i in range(0,len(nums)):
for j in range(x,nums[i]-1,-1):
dp[j] += dp[j-nums[i]]
return dp[x]