题目
给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
示例:
输入:“23”
输出:[“ad”, “ae”, “af”, “bd”, “be”, “bf”, “cd”, “ce”, “cf”].
说明:
尽管上面的答案是按字典序排列的,但是你可以任意选择答案输出的顺序。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/letter-combinations-of-a-phone-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
回溯
首先想到的是回溯算法,也就是之前“全排列”题目中接触到的解法,由于该题所给条件明确,只要出现就不会存在无效情况,于是需要列举每一种情况,可以把题目的解写成类似于树的形式。
学习借鉴
——队列
将输入的 digits 中第一个数字对应的每一个字母入队,然后将出队的元素与第二个数字对应的每一个字母组合后入队,直到遍历到 digits 的结尾,最后队列中的元素就是所求结果。
提交代码
回溯
class Solution:
def letterCombinations(self, digits: str) -> List[str]:
if not digits: return []
phone = {'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']}
def backtrack(conbination,nextdigit):
if len(nextdigit) == 0:
res.append(conbination)
else:
for letter in phone[nextdigit[0]]:
backtrack(conbination + letter,nextdigit[1:])
res = []
backtrack('',digits)
return res
- 时间复杂度:O(3M×4N)
M 是对应三个字母的数字个数,N 是对应四个字母的数字个数。 - 空间复杂度:O(3M×4N)
直接返回最后的结果,列表空间为组合数。
优化空间复杂度
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
- 空间复杂度:O(m+n)
其中 m 是输入中对应 3 个字母的数字个数,n 是对应 4 个字母的数字个数,m+n 是输入数字的总个数。
除了返回值以外,空间复杂度主要取决于哈希表以及回溯过程中的递归调用层数,哈希表的大小为常数,递归调用层数最大为 m+n。
队列
class Solution:
def letterCombinations(self, digits: str) -> List[str]:
if not digits: return []
phone = ['abc','def','ghi','jkl','mno','pqrs','tuv','wxyz']
queue = [''] # 初始化队列
for digit in digits:
for _ in range(len(queue)):
tmp = queue.pop(0)
for letter in phone[ord(digit)-50]:# 使用ASCII码定位
queue.append(tmp + letter)
return queue
学习总结
- 回溯算法
调用递归,在理解上还不是很到位,耗时较长; - 队列
将题目所给信息用另一种方式保存,利用ASCII码找到对应的索引,是一种新的思路。