leetcode练习——回溯算法（电话号码的字母组合）

最新推荐文章于 2022-11-28 10:29:39 发布

StriveQueen

最新推荐文章于 2022-11-28 10:29:39 发布

阅读量147

点赞数

文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/strivequeen/article/details/120564679

版权

LeetCode 专栏收录该内容

85 篇文章 0 订阅

订阅专栏

给定一个仅包含数字 2-9 的字符串，返回所有它能表示的字母组合。答案可以按任意顺序返回。
给出数字到字母的映射如下（与电话按键相同）。注意 1 不对应任何字母。

官方解法：https://leetcode-cn.com/problems/letter-combinations-of-a-phone-number/solution/dian-hua-hao-ma-de-zi-mu-zu-he-by-leetcode-solutio/

解法一：哈希表+回溯

（20ms/15.1MB）

class Solution:
    def letterCombinations(self, digits: str) -> List[str]:
        if not digits:
            return list()
        
        phoneMap = {
            "2": "abc",
            "3": "def",
            "4": "ghi",
            "5": "jkl",
            "6": "mno",
            "7": "pqrs",
            "8": "tuv",
            "9": "wxyz",
        }

        def backtrack(index: int):
            if index == len(digits):
                combinations.append("".join(combination))
            else:
                digit = digits[index]
                for letter in phoneMap[digit]:
                    combination.append(letter)
                    backtrack(index + 1)
                    combination.pop()

        combination = list()
        combinations = list()
        backtrack(0)
        return combinations

（28ms/15.1MB）

class Solution:
    def letterCombinations(self, digits: str) -> List[str]:
        if not digits:
            return list()
        
        phoneMap = {
            "2": "abc",
            "3": "def",
            "4": "ghi",
            "5": "jkl",
            "6": "mno",
            "7": "pqrs",
            "8": "tuv",
            "9": "wxyz",
        }

        groups = (phoneMap[digit] for digit in digits)
        return ["".join(combination) for combination in itertools.product(*groups)]
        # itertools.product(*iterables[, repeat]) # 对应有序的重复抽样过程，以元组的形式，根据输入的可遍历对象生成笛卡尔积，与嵌套的for循环类似。

repeat是一个关键字参数，指定重复生成序列的次数。