LeetCode 电话号码的字母组合

题目描述

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

题目解析

看到这个题目感觉应该用动态规划做（其实也不用动归，回溯就行，但是回溯的写法已经忘得差不多了），所以我自己就用空间换时间的方法来做。代码如下

#!/usr/bin/python3
# -*- coding:utf-8 -*-
# 电话号码的字母组合
class Solution:
    def combine(self,list1,list2):
        combinations = []
        for item in list1:
            for item2 in list2:
                combine = item + item2
                combinations.append(combine)

        return combinations

    def letterCombinations(self, digits):
        number_dict = {"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']}

        list_str = list(digits)
        if len(list_str) == 0:
            return []
        if len(list_str) == 1:
            return number_dict[list_str[0]]
        if (len(list_str) == 2):
            return self.combine(number_dict[list_str[0]],number_dict[list_str[1]])
        else:
            combinations = self.combine(number_dict[list_str[0]],number_dict[list_str[1]])
            for i in range(2,len(list_str)):
                combinations = self.combine(combinations,number_dict[list_str[i]])
            return combinations

sol = Solution()
print(sol.letterCombinations(""))

看着有点复杂，其实思想很简单，用一个combine函数来计算两个数字对应的字母组合，然后将当前的返回值存储在一个combination数组中。然后不断将combination和下一个数字对应的字母组合。时间复杂度 $(O^3)$，相比较一些指数级别的复杂度还将就吧。

Tips

感觉自己应该根据类型来刷题，比如：数组，动态规划，树等等，而不是想刷啥刷啥，看哪道题目顺眼就刷。