找出所有相加之和为 n
的 k
个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
思路:循环+递归,循环1-9个数字,用startindex记录循环的开始避免组合重复。递归处理startindex之后的数字。单层递归处理为将当前遍历的数字添加到path中,终止条件:如果path长度为k,则判断相加之和是否为n,如果是,则将path添加到结果集ans中,如果不是则不做处理。return。回溯为将path最后一个数值pop。综上,递归参数为:path,ans,startindex, cur_sum
剪枝优化:当当前和大于目标时,直接回溯。当startIndex后面的数字加上路径长度不可能满足k个数字时,直接回溯
Code(python):
class Solution:
def __init__(self):
self.path = []
self.ans = []
def combinationSum3(self, k: int, n: int) -> List[List[int]]:
self.trackback(k,n,1,0)
return self.ans
def trackback(self, k: int, n:int, startIndex:int, cur_sum:int):
if cur_sum > n:
return
if len(self.path) == k:
if cur_sum == n:
self.ans.append(self.path[:])
return
for i in range(startIndex, 9 + len(self.path) - k + 2):
cur_sum += i
self.path.append(i)
self.trackback(k, n, i + 1, cur_sum)
self.path.pop()
cur_sum -= i
给定一个仅包含数字 2-9
的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
思路:两层遍历+递归,遍历字符串和遍历数字对应的字符。因为给定字符串每一个字符都要用到,所以这里使用index来标记当前处理的字符。当index为字符串长度,则说明字符串遍历完毕,将path添加到ans中。
class Solution:
def __init__(self):
self.letterMap = [
"", # 0
"", # 1
"abc", # 2
"def", # 3
"ghi", # 4
"jkl", # 5
"mno", # 6
"pqrs", # 7
"tuv", # 8
"wxyz" # 9
]
def getCombinations(self, digits, index, path, result):
if index == len(digits):
result.append(''.join(path))
return
digit = int(digits[index])
letters = self.letterMap[digit]
for letter in letters:
path.append(letter)
self.getCombinations(digits, index + 1, path, result)
path.pop()
def letterCombinations(self, digits):
result = []
if len(digits) == 0:
return result
self.getCombinations(digits, 0, [], result)
return result