leetcode500键盘行python实现

1.问题描述

---

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

示例：

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

注意：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/keyboard-row
 

2.python实现

第一遍错误实现

class Solution(object):
    def findWords(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        one = ['q','w','e','r','t','y','u','i','o','p','Q','W','E','R','T','Y','U','I','O','P']
        two = ['a','s','d','f','g','h','j','k','l','A','S','D','F','G','H','J','K','L']
        three = ['z','x','c','v','b','n','m','Z','X','C','V','B','N','M']
        for word in words:
            flag = list()
            index = 0
            for c,i in zip(word,range(len(word))):
                if c in one:
                    flag.append(1)
                elif c in two:
                    flag.append(2)
                else:
                    flag.append(3)
                if i!=0 and flag[i-1]!=flag[-1]:
                    index = -1
                    break
            if index==-1:
                words.remove(word)
        return words
                
                    

第二遍正确的

class Solution(object):
    def findWords(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        one = ['q', 'w', 'e', 'r', 't', 'y', 'u', 'i', 'o', 'p', 'Q', 'W', 'E', 'R', 'T', 'Y', 'U', 'I', 'O', 'P']
        two = ['a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l', 'A', 'S', 'D', 'F', 'G', 'H', 'J', 'K', 'L']
        three = ['z', 'x', 'c', 'v', 'b', 'n', 'm', 'Z', 'X', 'C', 'V', 'B', 'N', 'M']
        re_list = list()
        for word in words:
            flag = list()
            index = 0
            for c, i in zip(word, range(len(word))):
                if c in one:
                    flag.append(1)
                elif c in two:
                    flag.append(2)
                else:
                    flag.append(3)
                if i != 0 and flag[i - 1] != flag[-1]:
                    index = -1
                    break
            if index == 0:
                re_list.append(word)
        return re_list

3.遇到的问题

for循环中关于列表list中remove的坑（特别注意）见链接 https://blog.csdn.net/qq_36387683/article/details/86659222