You are given a list of non-negative integers, a1, a2, ..., an, and a target, S. Now you have 2 symbols +
and -
. For each integer, you should choose one from +
and -
as its new symbol.
Find out how many ways to assign symbols to make sum of integers equal to target S.
Example 1:
Input: nums is [1, 1, 1, 1, 1], S is 3. Output: 5 Explanation: -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 There are 5 ways to assign symbols to make the sum of nums be target 3.
Note:
- The length of the given array is positive and will not exceed 20.
- The sum of elements in the given array will not exceed 1000.
- Your output answer is guaranteed to be fitted in a 32-bit integer.
------------------------------------------------------------------------------------
注意边界条件
class Solution:
def findTargetSumWays(self, nums: List[int], S: int) -> int:
l,c,n = len(nums),0,1
if (l <= 0):
return 0
layers = [{-nums[0]:1,nums[0]:1},{}] if (nums[0] != 0) else [{0:2},{}]
for i in range(1,l):
for kk,vv in layers[c].items():
for dk in [-nums[i],nums[i]]:
newk = kk+dk
newv = layers[n][newk] if newk in layers[n] else 0
layers[n][newk] = newv+vv
layers[c].clear()
c,n = n,c
return layers[c][S] if S in layers[c] else 0