题目
给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
示例:
输入:“23”
输出:[“ad”, “ae”, “af”, “bd”, “be”, “bf”, “cd”, “ce”, “cf”].
说明:
尽管上面的答案是按字典序排列的,但是你可以任意选择答案输出的顺序。
思路
把每个数字字符和与之对应的字母字符列表做成一个字典,然后先把digits中的第一个和第二个数字字符对应的字母字符列表依次遍历一边再相加组合起来,得到一个新的字母字符列表,再依次把这个字母字符列表与后一个数字字符对应的字母字符列表遍历组合相加,再得到一个新的字母字符列表,依次类推。
除此之外,对特殊情况做特殊处理,包括输入长度为0和1的字符串。
代码
class Solution:
def letterCombinations(self, digits):
label={'2':['a','b','c'],'3':['d','e','f'],'4':['g','h','i'],'5':['j','k','l'],\
'6':['m','n','o'],'7':['p','q','r','s'],'8':['t','u','v'],'9':['w','x','y','z']}
n=len(digits)
if n==0: #针对特殊情况处理
return []
if n==1:
return label[digits]
count=1
x=label[digits[0]] #初始化需要遍历的字母字符列表x和y
y=label[digits[count]]
while count<n:
record=[] #记录每次组合相加得到的字符串
for i in x:
for j in y:
record.append(i+j)
count+=1
x=record[:] #更行下一轮需要遍历的字母字符列表
if count<n:
y=label[digits[count]]
return x