给定一个单词列表,只返回可以使用在键盘同一行的字母打印出来的单词。键盘如下图所示。
示例:
输入: ["Hello", "Alaska", "Dad", "Peace"]
输出: ["Alaska", "Dad"]
注意:
你可以重复使用键盘上同一字符。
你可以假设输入的字符串将只包含字母。
class Solution:
def findWords(self, words: List[str]) -> List[str]:
a = ['q', 'w', 'e', 'r', 't', 'y', 'u', 'i', 'o', 'p',
'Q', 'W', 'E', 'R', 'T', 'Y', 'U', 'I', 'O', 'P']
b = ['a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l',
'A', 'S', 'D', 'F', 'G', 'H', 'J', 'K', 'L']
c = ['z', 'x', 'c', 'v', 'b', 'n', 'm',
'Z', 'X', 'C', 'V', 'B', 'N', 'M']
re = []
for each in words:
aa = 0
bb = 0
cc = 0
for i in each:
if i in a:
aa += 1
elif i in b:
bb += 1
elif i in c:
cc += 1
if aa == len(each) or bb == len(each) or cc == len(each):
re.append(each)
return re