给定一个单词列表,只返回可以使用在键盘同一行的字母打印出来的单词。键盘如下图所示。
输入: ["Hello", "Alaska", "Dad", "Peace"] 输出: ["Alaska", "Dad"]
- 你可以重复使用键盘上同一字符。
- 你可以假设输入的字符串将只包含字母。
class Solution:
def findWords(self, words: List[str]) -> List[str]:
result = []
row_q = set("qwertyuiop")
row_a = set("asdfghjkl")
row_z = set("zxcvbnm")
for word in words:
word_set = set(word.lower())
if word_set <= row_q or word_set <= row_a or word_set <= row_z:
result.append(word)
return result