[LeetCode in Python] 17 (M) letter combinations of a phone number 电话号码的字母组合

Posted 2021-03-04 ET民工[源自火星]

题目：

https://leetcode-cn.com/problems/letter-combinations-of-a-phone-number/

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

示例:

输入："23"
输出：["ad", "ae", "af", "bd", "be", "bf", "cd", "ce", "cf"].
说明:
尽管上面的答案是按字典序排列的，但是你可以任意选择答案输出的顺序。

解题思路

标准DFS套路，注意dfs()前后一行的处理。

代码

class Solution:
    def letterCombinations(self, digits: str) -> List[str]:
        # - sanity check
        if not digits:
            return []

        # - digit -> chars
        digit_dict = {
            '2': 'abc',
            '3': 'def',
            '4': 'ghi',
            '5': 'jkl',
            '6': 'mno',
            '7': 'pqrs',
            '8': 'tuv',
            '9': 'wxyz'
        }

        res = []

        # - dfs
        def dfs(index, s):
            # - return
            if index == len(digits):
                res.append(s)
                return

            # - for every branches
            for c in digit_dict[digits[index]]:
                s += c
                dfs(index+1, s)
                s = s[:-1]
        
        # - from digits[0]
        dfs(0, '')

        return res