需求
给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。
数字到字母的映射,与电话按键相同。注意 1 不对应任何字母。
示例
输入:"23"
输出:["ad", "ae", "af", "bd", "be", "bf", "cd", "ce", "cf"].
说明:尽管上面的答案是按字典序排列的,但是你可以任意选择答案输出的顺序。
方法一:遍历上次的组合结果,逐个合并
首先判断特殊情况,数字长度为0或数字为空时,直接返回空列表[];
基于数字和字母构建字典,遍历digits,获取数字对应的字母组合,遍历字母组合,同时遍历上次组合的结果,合并后添加到中间结果,一个数字遍历完成,对组合结果重新赋值为最新的中间结果;
特别地,初始化组合结果时,不能为空列表,如果为空列表,则不会执行遍历组合结果的代码段,本例中需要拼接字符,所以设置为一个空字符,对结果没有影响;
参考代码
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()
本方法的实现逻辑和方法一基本一致,主要区别是使用了python内置的追加元素的函数
extend()
,无需显式地对上次的组合结果进行遍历,就可以逐个(而不是作为整体)与字母拼接后,形成新的组合结果;由于digits的输入顺序,会影响最后的组合顺序,在遍历digits和对应的字母时,需要进行倒序遍历;
参考代码
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']