leetcode 500 键盘行

题目

给定一个单词列表，只返回可以使用在键盘同一行的字母打印出来的单词。键盘如下图所示。

 

 

示例：

输入: ["Hello", "Alaska", "Dad", "Peace"]
输出: ["Alaska", "Dad"]
 

注意：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/keyboard-row
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码

class Solution(object):
    def findWords(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        l1 = ['q', 'w', 'e', 'r', 't', 'y', 'u', 'i', 'o', 'p']
        l2 = ['a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l']
        l3 = ['z', 'x', 'c', 'v', 'b', 'n', 'm']
        res = []
        for word in words:
            _word = []
            for _w in list(word):
                _word.append(_w.lower())
            if not (set(_word) - set(l1)) or not (set(_word) - set(l2)) or not (set(_word) - set(l3)):
                res.append(word)
        return res