Leetcode 500. 键盘行 C++

Leetcode 500. 键盘行

题目

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

示例：

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

注意：

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

题解

我们利用哈希表记录字母对应的行，从而进行判别是否在同一行。然后遍历单词列表，只有所有字母在一行，才加到答案中。详细过程见代码

代码

vector<string> findWords(vector<string>& words) {
        unordered_map<char,int> index = {
            {'q',0},{'w',0},{'e',0},{'r',0},{'t',0},{'y',0},{'u',0},{'i',0},{'o',0},{'p',0},
            {'Q',0},{'W',0},{'E',0},{'R',0},{'T',0},{'Y',0},{'U',0},{'I',0},{'O',0},{'P',0},
            {'a',1},{'s',1},{'d',1},{'f',1},{'g',1},{'h',1},{'j',1},{'k',1},{'l',1},
            {'A',1},{'S',1},{'D',1},{'F',1},{'G',1},{'H',1},{'J',1},{'K',1},{'L',1},
            {'z',2},{'x',2},{'c',2},{'v',2},{'b',2},{'n',2},{'m',2},
            {'Z',2},{'X',2},{'C',2},{'V',2},{'B',2},{'N',2},{'M',2}
        };
        int n = words.size();
        vector<string> ans;
        for(int i=0; i<n; i++){
            int row = index[words[i][0]],j;
            for(j=1; j<words[i].length(); j++){
                if(index[words[i][j]] != row)   break;
            }
            if(j == words[i].length())  ans.push_back(words[i]);
        }
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/keyboard-row
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。