LeetCode-Python-377. 组合总和 Ⅳ

给定一个由正整数组成且不存在重复数字的数组，找出和为给定目标正整数的组合的个数。

示例:

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]