4.回溯算法-组合总和3.py

# # leetcode216
# 找出所有相加之和为n的k个数的组合，且满足下列条件：
# 只使用数字1到9每个数字最多使用一次 返回所有可能的有效组合的列表 。
# 该列表不能包含相同的组合两次，组合可以以任何顺序返回。
#
#
#
# 示例1:
# 输入: k = 3, n = 7
# 输出: [[1, 2, 4]]
# 解释:
# 1 + 2 + 4 = 7
# 没有其他符合的组合了。
#
# 示例2:
# 输入: k = 3, n = 9
# 输出: [[1, 2, 6], [1, 3, 5], [2, 3, 4]]
# 解释:
# 1 + 2 + 6 = 9
# 1 + 3 + 5 = 9
# 2 + 3 + 4 = 9
# 没有其他符合的组合了。
# 示例
# 3:
#
# 输入: k = 4, n = 1
# 输出: []
# 解释: 不存在有效的组合。
# 在[1, 9]
# 范围内使用4个不同的数字，我们可以得到的最小和是1 + 2 + 3 + 4 = 10，因为10 > 1，没有有效的组合。


k = 4
n = 1

def combinations(start_index, k, n, path,res):
    if sum(path) == n and len(path) == k:
        res.append(path[:])

    for i in range(start_index,10):
        path.append(i)
        combinations(i+1, k, n, path, res)
        path.pop()
    return res
path = []
res = []
print(combinations(1, k, n, path,res))