LeetCode刷题实战500：键盘行

栈长 • 2022年1月17日 13:52 • 编程技术 • 阅读 409

算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，这个公众号后续每天带大家做一道算法题，题目就从LeetCode上面选！

今天和大家聊的问题叫做 键盘行，我们先来看题面：

https://leetcode-cn.com/problems/keyboard-row/

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

美式键盘中：

第一行由字符 “qwertyuiop” 组成。

第二行由字符 “asdfghjkl” 组成。

第三行由字符 “zxcvbnm” 组成。

示例

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

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

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

解题

https://leetcode-cn.com/problems/keyboard-row/solution/jian-pan-xing-by-leetcode-solution-bj5e/

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

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

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

class Solution {    
    public String[] findWords(String[] words) {
        List<String> list = new ArrayList<String>();
        String rowIdx = "12210111011122000010020202";
        for (String word : words) {
            boolean isValid = true;
            char idx = rowIdx.charAt(Character.toLowerCase(word.charAt(0)) - 'a');
            for (int i = 1; i < word.length(); ++i) {
                if (rowIdx.charAt(Character.toLowerCase(word.charAt(i)) - 'a') != idx) {
                    isValid = false;
                    break;
                }
            }
            if (isValid) {
                list.add(word);
            }
        }
        String[] ans = new String[list.size()];
        for (int i = 0; i < list.size(); ++i) {
            ans[i] = list.get(i);
        }
        return ans;
    }
}