377. 组合总和IV

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

示例:

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。

思路: 看到这个题目，有两种思路，一个是动态规划，一个递归。这里只需要求出组合个数，不需要罗列出所有的组合，所以动态规划即可。

有几个需要注意的地方

• 题目中 不存在重复的数字 有什么用途?
• 如何控制顺序不同的序列 ?
• 怎么写状态转移方程?

先思考一下 状态转移方程:
我们直接把题目要求的当做我们的状态dp[i]: 和为i的组合个数。

• 第一个数选择 nums[0], 那么有dp[i - nums[0]]个组合个数
• 第一个数选择 nums[1], 那么有dp[i - nums[1]]个组合个数
• 第一个数选择 nums[2], 那么有dp[i - nums[2]]个组合个数
  …

因此对于dp[i], 枚举所有nums中的数num作为第一个数,则有

dp[i] = sum(dp[i - num]), num in nums, 而且 num <= i

初始条件: dp[0] = 1， 即和为0的组合个数，有1个: 空。

这样构造的状态转移方程的属性满足了"顺序不同的序列"的要求。

至于不存在重复的数字，因为每个数字可以任意多次的选择，所以这个问题不重要。

程序如下:

class Solution:
    def combinationSum4(self, nums: List[int], target: int) -> int:
        dp = [0] * (target + 1)
        dp[0] = 1
        for i in range(1, target + 1):
            for num in nums:
                if num <= i:
                    dp[i] += dp[i - num]
        return dp[target]

最初写了如下的程序:

# 超时, 通过 16 / 17 个测试用例
class Solution:
    def combinationSum4(self, nums: List[int], target: int) -> int:
        dp = [0] * (target + 1)
        dp[0] = 1
        for i in range(1, target + 1):
            for j in range(1, i+1):
                if j in nums:
                    dp[i] += dp[i - j]
        return dp[target]

这样写有两个坏处:

• 需要遍历每一个1 - i之间的数，有些情况会比nums多遍历很多次
• 每次遍历 1 - i之间的数，都需要在nums寻找是否存在，这个会使程序减慢