力扣刷题学习500. 键盘行（C++）

题目描述

给你一个字符串数组 words ，只返回可以使用在 美式键盘 同一行的字母打印出来的单词。
美式键盘 中：
第一行由字符 “qwertyuiop” 组成。
第二行由字符 “asdfghjkl” 组成。
第三行由字符 “zxcvbnm” 组成。

测试用例

示例一：

输入：words = [“Hello”,“Alaska”,“Dad”,“Peace”]
输出：[“Alaska”,“Dad”]

示例二：

输入：words = [“omk”]
输出：[]

示例三：

输入：words = [“adsdf”,“sfd”]
输出：[“adsdf”,“sfd”]

提示：

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

解答

思路：
首先我想到的是建立三个数组用于存储每个字符，然后每次选择words中的一个字符串，开始一次比较数组中的字符。
随后对思路进行优化，先对键盘字母预处理，标记出它们的行数，这样只需用到一个数组，且比较效率更高。
其中还可以对字符串预处理，将大写转换为小写。

class Solution {
public:
    vector<string> findWords(vector<string>& words) {
        vector<string> ans;
        string rowIdx = "12210111011122000010020202";
        for (auto & word : words) {
            bool isValid = true;
            char idx = rowIdx[tolower(word[0]) - 'a'];
            for (int i = 1; i < word.size(); ++i) {
                if(rowIdx[tolower(word[i]) - 'a'] != idx) {
                    isValid = false;
                    break;
                }
            }
            if (isValid) {
                ans.emplace_back(word);
            }
        }
        return ans;
    }
};

复杂度分析

• 时间复杂度：O(L)O(L)，其中 LL 表示 \textit{words}words 中所有字符串的长度之和。
• 空间复杂度：O©O©，其中 CC 表示英文字母的个数，在本题中英文字母的个数为 2626。注意返回值不计入空间复杂度。