面试题 16.20. T9键盘（中等）- LeetCode

题目描述

自己解法

class Solution:
    def getValidT9Words(self, num: str, words: List[str]) -> List[str]:
        keyboard = {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']}
        equal = True
        ans = []
        
        for i,str in enumerate(words):
            equal = True
            for j,char in enumerate(str):
                if char not in keyboard[int(num[j])]:
                    equal = False
                    break
            if equal:
                ans.append(str)

        return ans 

题解区解法

列表生成器的剪枝

class Solution:
    def getValidT9Words(self, num: str, words: List[str]) -> List[str]:
        kb = {'2':'abc', '3':'def', '4':'ghi', '5':'jkl', '6':'mno', '7':'pqrs', '8':'tuv', '9':'wxyz'}
        ns = list(num)
        candidate = words
        for i, n in enumerate(ns):
            candidate = [w for w in candidate if w[i] in kb[n]]

        return candidate

数组求解

class Solution:
    def getValidT9Words(self, num: str, words: List[str]) -> List[str]:
        ans = list()
        arr = ['2','2','2','3','3','3','4','4','4','5','5','5','6','6','6','7','7','7','7','8','8','8','9','9','9','9']
        for word in words:
            index = 0
            flag = True
            for c in word:
                if num[index] == arr[ord(c) - ord('a')]:
                    index += 1
                else:
                    flag = False
                    break
            if flag: ans.append(word)
        return ans