One line of keyboard

class Solution:
# Parameters words For a list of strings
# Returns a list of strings
def findWords(self, words):
res = []
s = [“qwertyuiop”, “asdfghjkl”, “zxcvbnm”]
for w in words:
for j in range(3):
flag = 1
for i in w:
if i.lower() not in s[j]:
flag = 0
break
if flag == 1:
res.append(w)
break
return res
# The main function
if name == ‘main’:
word = [“Hello”, “Alaska”, “Dad”, “Peace”]
s = Solution()
print(“ Input is ：”,word)
print(“ Output is ：”,s.findWords(word))