LeetCode-500 键盘行

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

示例:

输入: ["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
发布了163 篇原创文章 · 获赞 14 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_24502469/article/details/102859991