题目描述:给定一个非负整数数组,a1, a2, …, an, 和一个目标数,S。现在你有两个符号 + 和 -。对于数组中的任意一个整数,你都可以从 + 或 -中选择一个符号添加在前面。
返回可以使最终数组和为目标数 S 的所有添加符号的方法数。
解题思路一:动态规划,遍历所有可以取到的值,通过对当前值进行加和减进行转移,代码如下:
class Solution:
def findTargetSumWays(self, nums: List[int], S: int) -> int:
n = len(nums)
total = sum(nums)
dp = [[0]*(2*total+1) for _ in range(n+1)]
dp[0][total] = 1
for i in range(1, n+1):
for j in range(2*total+1):
if j- nums[i-1] >= 0:
dp[i][j] += dp[i-1][j-nums[i-1]]
if j+nums[i-1] <= 2 * total:
dp[i][j] += dp[i-1][j+nums[i-1]]
return dp[-1][S+total] if total >= S else 0
解题思路二:枚举所有可能情况,结果等于S就+1
,用递归进行遍历,python会时间超时,代码如下:
class Solution:
count = 0
def calculate(self, nums, i, total, S):
if i == len(nums):
if total == S:
self.count += 1
else:
self.calculate(nums, i+1, total+nums[i], S)
self.calculate(nums, i+1, total-nums[i], S)
def findTargetSumWays(self, nums: List[int], S: int) -> int:
# count = 0
self.calculate(nums, 0, 0, S)
return self.count
解题思路三:转化为背包问题,把所有符号为正的数总和设为一个背包的容量,容量为x;把所有符号为负的数总和设为一个背包的容量,容量为y。在给定的数组中,有多少种选择方法让背包装满。令sum为数组的总和,则x+y = sum。而两个背包的差为S,则x-y=S。从而求得x=(S+sum)/2。基于上述分析,题目转换为背包问题:给定一个数组和一个容量为x的背包,求有多少种方式让背包装满。代码如下:
class Solution:
def findTargetSumWays(self, nums: List[int], S: int) -> int:
total = sum(nums)
if total < S or (S + total) % 2: return 0
target = (S + total) // 2
dp = [0] * (target + 1)
dp[0] = 1
for num in nums:
for i in range(target, num-1, -1):
dp[i] += dp[i-num]
return dp[-1]