代码随想录 回溯Ⅱ

216. 组合总和 III

找出所有相加之和为 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
        

17. 电话号码的字母组合

给定一个仅包含数字 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