代码随想录算法训练营第25天｜216.组合总和III 17.电话号码的字母组合

最新推荐文章于 2024-10-01 04:50:10 发布

小周小周考虑不周

最新推荐文章于 2024-10-01 04:50:10 发布

阅读量127

点赞数

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/dieforcpp/article/details/129217991

版权

216.组合总和III

题目链接：LeetCode - The World's Leading Online Programming Learning Platform

题目链接/文章讲解：代码随想录

视频讲解：和组合问题有啥区别？回溯算法如何剪枝？| LeetCode：216.组合总和III_哔哩哔哩_bilibili

设置一个stack path 一个结果res和一个当前和

定义回溯（n,k,start_index 传入start_index的原因是在回溯时我们需要知道他应该从哪一个起始值进行递归），首先设置basic condition 如果path长度为k时，把path放进res。如果不合符条件的话就进入回溯中的for loop，从start_index到 9-（k-len(path))+1 循环，将当前i append进path 并加上sum 做回溯递归，注意此时start_index是i+1，是从我们当前i的后一位递归。当递归到底返回后要注意回溯也就是pop我们的path并减去当前值。

最后执行我们的function并return我们res

也要注意剪枝如果sum大于n或者path长度大于k时直接返回。

class Solution(object):
    def __init__(self):
        self.path=[]
        self.res=[]
        self.sum_ = 0
        
    def combinationSum3(self, k, n):
        """
        :type k: int
        :type n: int
        :rtype: List[List[int]]
        """
        def backtracking(k,n,start_index):
            if self.sum_ > n or len(self.path) > k:
                return
            if len(self.path) == k:
                if self.sum_ == n:
                    self.res.append(self.path[:])
                return
            for i in range(start_index,10 - (k - len(self.path)) + 1):
                self.path.append(i)
                self.sum_ += i
                backtracking(k,n,i+1)
                self.sum_ -= i
                self.path.pop()
        backtracking(k,n,1)
        return self.res

17.电话号码的字母组合

题目链接：LeetCode - The World's Leading Online Programming Learning Platform

题目链接/文章讲解：代码随想录

视频讲解：还得用回溯算法！| LeetCode：17.电话号码的字母组合_哔哩哔哩_bilibili

解题思路：

先给方程设置init，一个当前path 一个结果res和一个letter_map

然后判断cornercase是否有digits。

设置一个递归方程，base case为如果当前path长度等于digits长度的话就可以将当前path放进res。

首先设置letters为当前digit对应map中的字母进行遍历，首先放入path，进行递归下一个digit，结束后我们要注意回溯将之前放入path的字母删除。递归结束后我们得到结果。

class Solution(object):
    def __init__(self):
        self.path=""
        self.res=[]
        self.letter_map = {
            '2': 'abc',
            '3': 'def',
            '4': 'ghi',
            '5': 'jkl',
            '6': 'mno',
            '7': 'pqrs',
            '8': 'tuv',
            '9': 'wxyz'
        }
    def letterCombinations(self, digits):
        """
        :type digits: str
        :rtype: List[str]
        """
        if not digits:
            return 
        def backtracking(digits,start_index):
            if len(self.path)==len(digits):
                self.res.append(self.path[:])
                return
            letters = self.letter_map[digits[start_index]]
            for letter in letters:
                self.path += letter
                backtracking(digits,start_index + 1)
                self.path=self.path[:-1]
        backtracking(digits,0)
        return self.res