500. 键盘行

这是我参与11月更文挑战的第1天，活动详情查看：2021最后一次更文挑战

500. 键盘行

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

美式键盘 中：

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

示例 1：

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

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

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

提示：

• 1 <= words.length <= 20
• 1 <= words[i].length <= 100
• words[i] 由英文字母（小写和大写字母）组成

解题思路

1. 使用3个set分别记录键盘每一行的所有字符，对每个字符串判断，是否字符串的每个字符都属于属于键盘的同一行
2. 我们还可以对键盘的每一行进行分组编码，分为3组，每个字符根据在键盘的位置归到对应的组别，如果一个字符串是由键盘的同一行单词组成，那么必然它每个字符所对应的组别都是相同的。

代码

class Solution {
public:
    vector<string> findWords(vector<string> &words) {

        vector<string> res;
        unordered_set<char> set1{'q', 'w', 'e', 'r', 't', 'y', 'u', 'i', 'o', 'p'};
        unordered_set<char>set2{'a', 's', 'd', 'f', 'g', 'h','j', 'k', 'l'};
        unordered_set<char>set3{'z', 'x','c', 'v','b', 'n','m'};

        for (string s:words) {
            int i = 0;
            for (; i < s.length(); ++i) {
                if (set1.find(tolower(s[i])) == set1.end())
                    break;
            }
            if (i == s.length()) {
                res.push_back(s);
                continue;
            }
            i = 0;
            for (; i < s.length(); ++i) {
                if (set2.find(tolower(s[i])) == set2.end())
                    break;
            }
            if (i == s.length()) {
                res.push_back(s);
                continue;
            }
            i = 0;
            for (; i < s.length(); ++i) {
                if (set3.find(tolower(s[i])) == set3.end())
                    break;
            }
            if (i == s.length()) {
                res.push_back(s);
                continue;
            }

        }
        return res;
    }
};

class Solution {

public:

    vector<string> findWords(vector<string> &words) {

        vector<string> res;

        string rowIdx = "12210111011122000010020202";

        for (auto s:words) {

            bool flag = true;

            char group = rowIdx[tolower(s[0]) - 'a'];

            for (int i=1;i<s.size();i++) {

                if (rowIdx[tolower(s[i]) - 'a'] != group) {

                    flag = false;

                    break;
                }
            }

            if (flag)
                res.emplace_back(s);
        }
        return res;
    }
};

\

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

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