一.题目描述
给定一个非负整数数组,a1, a2, ..., an, 和一个目标数,S。现在你有两个符号 + 和 -。对于数组中的任意一个整数,你都可以从 + 或 -中选择一个符号添加在前面。
返回可以使最终数组和为目标数 S 的所有添加符号的方法数。
示例 1:
输入: nums: [1, 1, 1, 1, 1], S: 3
输出: 5
解释:
-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
一共有5种方法让最终目标和为3。
注意:
数组的长度不会超过20,并且数组中的值全为正数。
初始的数组的和不会超过1000。
保证返回的最终结果为32位整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/target-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二.思路
每个数字前面有正负两种情况,总共2^n可能
使用动态规划,将当前状态分为两种情况:当前元素前面为正号或者负号,将当前需要满足的和作为当前状态
递推关系式:
DP[i][j]=DP[i-1][j-nums[i]]+DP[i-1][j+nums[i]]
DP[i][j]表示0-i个元素需要满足和为j的所有可能性
当nums[i]前面为负号时,当前情况的可能性跟0 to i-1的元素中满足和为j+nums[i]的情况数一样
当nums[i]为正号时也是一样的
需要注意第一个值为0 的情况,每个元素的需要求解的和的范围也是nums[0:i] to -nums[0:i]
代码如下:
class Solution:
def findTargetSumWays(self, nums: List[int], S: int) -> int:
Len=len(nums)
Sum=sum(nums)
if S>Sum:
return 0
Index1=[0 for i in range(S-Sum,S+Sum+1)]
if nums[0]==0:
Index1[nums[0]]=2
else:
Index1[-nums[0]]=1
Index1[nums[0]]=1
for i in range(1,Len):
Index2=[0 for i in range(S-Sum,S+Sum+1)]
for j in range(-sum(nums[:i+1]),sum(nums[:i+1])+1):
Index2[j]=Index1[j-nums[i]]+Index1[j+nums[i]]
Index1=copy.deepcopy(Index2)
return Index1[S]