力扣每日一题500.键盘行（来学习代码优化）

给你一个字符串数组 words ，只返回可以使用在 美式键盘 同一行的字母打印出来的单词。键盘如下图所示。

美式键盘 中：

第一行由字符 "qwertyuiop" 组成。
第二行由字符 "asdfghjkl" 组成。
第三行由字符 "zxcvbnm" 组成。

示例 1：

输入：words = ["Hello","Alaska","Dad","Peace"]
输出：["Alaska","Dad"]
示例 2：

输入：words = ["omk"]
输出：[]
示例 3：

输入：words = ["adsdf","sfd"]
输出：["adsdf","sfd"]

直接给出自己写的暴力的垃圾代码，让大家喷喷（来找出优化方案）：

class Solution:
    def findWords(self, words: List[str]) -> List[str]:
        l1=['q','w','e','r','t','y','u','i','o','p']
        l2=['a','s','d','f','g','h','j','k','l']
        l3=['z','x','c','v','b','n','m']
        l=[]
        for i in words:
            m=i
            i=i.lower()
            k=0
            if i[k] in l1:
                k=k+1
                for j in range (len(i)):
                    if i[j] not in l1:
                        break
                    if j==len(i)-1:
                        l.append(m)
            if k==1:
                continue
            if i[k] in l2:
                k=k+1
                for j in range (len(i)):
                    if i[j] not in l2:
                        break
                    if j==len(i)-1:
                        l.append(m)
            if k==1:
                continue
            if i[k] in l3:
                k=k+1
                for j in range (len(i)):
                    if i[j] not in l3:
                        break
                    if j==len(i)-1:
                        l.append(m)  
        return l

这个代码思路很清晰，也很暴力，就是字符转小写，判断每个单词的每个字母是否在同一行，其实如果可以写外部函数就是一个函数调用三次就行，其实也很便捷，一看时间复杂度，空间复杂度都是o(n),所以肯定是垃圾代码。不过怎么优化呢？

一般想优化，都是找规律的那种思路，而不是一心暴力求解。

看一个官方解：

class Solution:
    def findWords(self, words: List[str]) -> List[str]:
        ans = []
        rowIdx = "12210111011122000010020202"
        for word in words:
            idx = rowIdx[ord(word[0].lower()) - ord('a')]
            if all(rowIdx[ord(ch.lower()) - ord('a')] == idx for ch in word):
                ans.append(word)
        return ans

思路：

我们为每一个英文字母标记其对应键盘上的行号，然后检测字符串中所有字符对应的行号是否相同。

我们可以预处理计算出每个字符对应的行号。

遍历字符串时，统一将大写字母转化为小写字母方便计算。

复杂度分析

时间复杂度：O(L)O(L)，其中 LL 表示 \textit{words}words 中所有字符串的长度之和。

空间复杂度：O(C)O(C)，其中 CC 表示英文字母的个数，在本题中英文字母的个数为 2626。注意返回值不计入空间复杂度。

代码量和复杂度都下降很多，其实这道题也就是简单题，官方给出的解，以及评论区的都有点遍历完的意思，都没有很亮眼的解，不知道有没有大佬能出一个更牛给我looklook。