49/300
目标和 给定一个非负整数数组,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位整数。动态规划类型,第六道: 暴力全算,O(2n),然后数目标和有几个。 动态规划pass 下面这种方法算是暴力的优化,{result: frequency}, 也就是说遇到重复的值,直接合并,加频次。然后提取目标和的频次就好。
举例 [1,1,1],画个丑图,配合代码食用:
python 字典查找指定键值的两种办法 快速上手defaultdict 避免KeyError collections.defaultdict() dict.get(key, default=None)
class Solution: def findTargetSumWays(self, nums, S): #initialize {0:1} mydict = collections.defaultdict(int) mydict[0] = 1 for num in nums: #initialize {} innerdict = collections.defaultdict(int) for prev_sum in mydict: #add positive innerdict[prev_sum+num] += mydict[prev_sum] #add negative innerdict[prev_sum-num] += mydict[prev_sum] #prepare for the next mydict = innerdict return mydict[S]