LeetCode_Python(17)_电话号码的字母组合

需求

给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。

数字到字母的映射,与电话按键相同。注意 1 不对应任何字母。


LeetCode_Python(17)_电话号码的字母组合_第1张图片

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

方法一:遍历上次的组合结果,逐个合并

  1. 首先判断特殊情况,数字长度为0或数字为空时,直接返回空列表[];

  2. 基于数字和字母构建字典,遍历digits,获取数字对应的字母组合,遍历字母组合,同时遍历上次组合的结果,合并后添加到中间结果,一个数字遍历完成,对组合结果重新赋值为最新的中间结果;

  3. 特别地,初始化组合结果时,不能为空列表,如果为空列表,则不会执行遍历组合结果的代码段,本例中需要拼接字符,所以设置为一个空字符,对结果没有影响;

  4. 参考代码

def num_letter_combinations(digits):
    if len(digits) == 0 or digits == '': return []

    num_letter = {
    '2': 'abc',
    '3': 'def',
    '4': 'ghi',
    '5': 'jkl',
    '6': 'mno',
    '7': 'pqrs',
    '8': 'tuv',
    '9': 'wxyz'
    }   
    res = ['']

    for d in digits:
        if d in num_letter.keys():
            tmp = []
            for s in num_letter[d]:
                for r in res:
                    tmp.append(s + r)
            res = tmp
    return res 

digits = '23'
print(num_letter_combinations(digits))
['da', 'db', 'dc', 'ea', 'eb', 'ec', 'fa', 'fb', 'fc']


方法二:倒序遍历 + extend()

  1. 本方法的实现逻辑和方法一基本一致,主要区别是使用了python内置的追加元素的函数extend(),无需显式地对上次的组合结果进行遍历,就可以逐个(而不是作为整体)与字母拼接后,形成新的组合结果;

  2. 由于digits的输入顺序,会影响最后的组合顺序,在遍历digits和对应的字母时,需要进行倒序遍历;

  3. 参考代码

def num_letter_combinations(digits):
    if len(digits) == 0 or digits == '': return []

    num_letter = {
    '2': 'abc',
    '3': 'def',
    '4': 'ghi',
    '5': 'jkl',
    '6': 'mno',
    '7': 'pqrs',
    '8': 'tuv',
    '9': 'wxyz'
    } 
    res = ['']
    for d in digits:
        if d not in num_letter.keys(): return []
        else:
            tmp = []
            for s in num_letter[d]:
                tmp.extend(s + x for x in res)
            res = tmp
    return res
                    

digits = '23'
print(num_letter_combinations(digits))
['da', 'db', 'dc', 'ea', 'eb', 'ec', 'fa', 'fb', 'fc']

你可能感兴趣的:(LeetCode_Python(17)_电话号码的字母组合)