给定一个由正整数组成且不存在重复数字的数组,找出和为给定目标正整数的组合的个数。
示例:
nums = [1, 2, 3] target = 4 所有可能的组合为: (1, 1, 1, 1) (1, 1, 2) (1, 2, 1) (1, 3) (2, 1, 1) (2, 2) (3, 1) 请注意,顺序不同的序列被视作不同的组合。 因此输出为 7。
进阶:
如果给定的数组中含有负数会怎么样?
问题会产生什么变化?
我们需要在题目中添加什么限制来允许负数的出现?
致谢:
特别感谢 @pbrother 添加此问题并创建所有测试用例。
第一种思路:
DFS。
只能过一半CASE,会超时。
class Solution(object):
def combinationSum4(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
self.res = 0
def dfs(nums, target):
if target == 0:
self.res += 1
return
for num in nums:
if num > target:
break
dfs(nums, target - num)
nums.sort()
dfs(nums, target)
return self.res
第二种思路:
观察了一下,递归地方法会重复计算很多次,所以可以用dp来储存之前的结果,节省计算的时间。
dp[i]表示nums有多少种组成 i 的组合。
dp[0] = 1, 即什么都不选
dp[i] = dp[i - nums[0]] + dp[i - nums[1]] + dp[i - nums[2]] + .... + dp[i - nums[-1]]
class Solution(object):
def combinationSum4(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
dp = [0 for _ in range(target + 1)]
dp[0] = 1
nums.sort()
for i in range(1, target + 1):
for num in nums:
if i >= num:
dp[i] += dp[i - num]
return dp[target]