LeetCode 500. 键盘行

500. 键盘行

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

示例1:

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

注意:

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

解：

class Solution:
    def findWords(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        p = ['qwertyuiop', 'asdfghjkl', 'zxcvbnm']
        ends = []

        for i in p:
            for word in words:
                temp = set(word.lower())
                if temp.issubset(i):
                    ends.append(word)
        return ends