算法刷题打卡022 | 回溯2

最新推荐文章于 2024-09-06 10:38:06 发布

tsy_

最新推荐文章于 2024-09-06 10:38:06 发布

阅读量120

点赞数

分类专栏：刷题文章标签：算法 leetcode

本文链接：https://blog.csdn.net/tsy_0827/article/details/129461631

版权

刷题专栏收录该内容

49 篇文章 0 订阅

订阅专栏

文章介绍了如何使用回溯算法解决LeetCode中的两道题目：组合总和III要求找到和为n的所有整数组合，关键在于限制下一层遍历的范围；电话号码的字母组合则涉及到数字与字母的映射，同样利用回溯生成所有可能的字符串组合。

摘要由CSDN通过智能技术生成

LeetCode 216 组合总和III

题目链接：216. 组合总和 III - 力扣（Leetcode）

这道题和77 组合解题思路完全相同，只是收集结果时多加了和为n的限制。具体代码实现时一度出现重复使用数字的情况，debug发现是for循环中调用递归函数时，每次遍历起始数字start传入了start+1，导致下一层遍历的起始位置有出入（可以获得已经遍历过的值），正确做法应该是传入当前i的下一个数字，限制下一层可遍历的范围：

class Solution:
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        result, path = [], []
        def backtracking(start, remain):
            if remain < 0:
                return
            if len(path) == k:
                if remain == 0:
                    result.append(path[:])
                return 
            end = 9 - (k - len(path)) + 1
            for i in range(start, end + 1):
                path.append(i)
                backtracking(i+1, remain-i)
                path.pop()
        
        backtracking(1, n)
        return result

LeetCode 17 电话号码的字母组合

题目链接：17. 电话号码的字母组合 - 力扣（Leetcode）

这道题的回溯本质上和数字的组合没什么不同，只是组合形式多了一个映射关系。递归函数传入的idx是当前遍历到的数字下标，for循环遍历的就是当前数字对应的字母序列，深度优先遍历使得函数一直递归到digits最后一个数字才开始收集结果，不断回溯，最终得到所有组合结果：

class Solution:
    def letterCombinations(self, digits: str) -> List[str]:
        if len(digits) == 0:
            return []
        digit_map = {'2':'abc', '3':'def', '4': 'ghi', '5':'jkl', '6':'mno', '7':'pqrs', '8':'tuv', '9':'wxyz'}
        path, res = [] , []

        def backtracking(idx):
            if idx == len(digits):
                res.append(''.join(path.copy()))
                return
            chs = digit_map[digits[idx]]
            for ch in chs:
                path.append(ch)
                backtracking(idx + 1)
                path.pop()
        
        backtracking(0)
        return res

今天的回溯就先复习到这吧。