找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii
例:
输入: k = 3, n = 7 输出: [[1,2,4]] 解释: 1 + 2 + 4 = 7 没有其他符合的组合了
解析:
因为没有顺序要求,而且每个值要求不同,所以可以升序查找,即后面的值一定大于前面的值,那么这样就简单了,一个结果列表,一个临时列表,利用回溯法,在求解的过程中使用临时列表,当列表中的值满足条件时将值存入结果列表即可。
class Solution(object):
def combinationSum3(self, k, n):
"""
:type k: int
:type n: int
:rtype: List[List[int]]
"""
ans = [] # 结果列表
def dfs(arr, count, pos): # 参数分别为:临时列表;当前综合;需要判断的值
if count == n and len(arr) == k: # 满足条件
ans.append(arr) # 将满足条件的情况加入到结果列表
return
if count > n or len(arr) > k: # 剪枝
return
for i in range(pos, 10): # 在大于当前值得范围中继续查找
dfs(arr+[i], count+i, i+1)
dfs([], 0, 1) #调用函数
return ans