LeetCode系列: 500.键盘行

500.键盘行

https://leetcode-cn.com/problems/keyboard-row/
给定一个单词列表，只返回可以使用在键盘同一行的字母打印出来的单词。键盘如下图所示。

American keyboard

示例：

输入: [“Hello”, “Alaska”, “Dad”, “Peace”]
输出: [“Alaska”, “Dad”]

注意：

你可以重复使用键盘上同一字符。
你可以假设输入的字符串将只包含字母。

解题思路：

1.查表打法
2.注意大小写字母的差异
3.遇到不相同的就可以停止

代码实现：

class Solution(object):
    def findWords(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        
        
        res = []
        for word in words:
            if self.isOneLine(word):
                res.append(word)
        return res
            
    def isOneLine(self, word):
        letterMap = {
            'q': 1, 'w': 1, 'e': 1, 'r': 1,
            't': 1, 'y': 1, 'u': 1, 'i': 1, 
            'o': 1, 'p': 1, 'a': 2, 's': 2,
            'd': 2, 'f': 2, 'g': 2, 'h': 2,
            'j': 2, 'k': 2, 'l': 2, 'z': 3,
            'x': 3, 'c': 3, 'v': 3, 'b': 3,
            'n': 3, 'm': 3  
        }
        num = letterMap[word[0].lower()]
        for tmp in word:
            if letterMap[tmp.lower()] != num:
                return False
        return True