给定一个由正整数组成且不存在重复数字的数组,找出和为给定目标正整数的组合的个数。
示例:
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。
方案:
(1)首先考虑回溯方法,如下:【结果超时】
class Solution:
def combinationSum4(self, nums: List[int], target: int) -> int:
result = []
def dp(path):
if path not in result and sum(path)==target:
result.append(path[:])
if path in result or sum(path) > target:
return
for i in nums:
path.append(i)
dp(path)
path.pop()
dp([])
return len(result)
(2)动态规划,参考题解。
class Solution:
def combinationSum4(self, nums: List[int], target: int) -> int:
if len(nums) == 0 or target <= 0:
return 0
dp = [0] * (target + 1)
dp[0] = 1
# 这一步很关键,想想为什么 dp[0] 是 1
# 因为 0 表示空集,空集和它"前面"的元素凑成一种解法,所以是 1
# 这个值被其它状态参考,设置为 1 是合理的
for i in range(1, target + 1):
for j in nums:
if i >= j:
dp[i] += dp[i-j]
return dp[-1]
由于不需要返回具体的组合情况,及重复计算地方较多,故采用 “动态规划” 的方法求解。类似斐波那契数列: