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

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

            

你可能感兴趣的:(算法,leetcode,数据结构)