500. Keyboard Row

逗逗军001

于 2017-08-12 19:03:00 发布

阅读量278

点赞数

CC 4.0 BY-SA版权

分类专栏：每天一Leetcode 文章标签： leetcode c++

本文链接：https://blog.youkuaiyun.com/u010956473/article/details/77127031

每天一Leetcode 专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一种使用C++实现的算法，该算法能够找出所有仅由键盘同一行字符组成的单词。通过遍历输入单词列表并检查每个单词的字母是否位于预定义的键盘行集合之一中来完成这一任务。

一、题目：

Given a List of words, return the words that can be typed using letters of alphabet on only one row's of American keyboard like the image below.

这里写图片描述

二、解题

class Solution {
    /*
    从第一个单词开始找，先找出该单词首字母在哪行，
    然后看那个单词是否都在那行。
    若在，给添加到result中。

    */
public:
    vector<string> findWords(vector<string>& words) {
        unordered_set<char> row1{ 'q', 'w', 'e', 'r', 't', 'y','u', 'i', 'o', 'p' };
        unordered_set<char> row2{ 'a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l' };
        unordered_set<char> row3{ 'z', 'x', 'c', 'v', 'b' ,'n', 'm' };
        vector<unordered_set<char>> rows{ row1, row2, row3 };

        vector<string> validWords;
        for (int i = 0; i<words.size(); ++i) {
            int row = 0;

            for (int k = 0; k<3; ++k) {
                if (rows[k].count((char)tolower(words[i][0])) > 0) row = k;
            }

            for (int j = 0; j<words[i].size(); ++j) {
                if (rows[row].count((char)tolower(words[i][j])) == 0)
                    break;
                if (j == words[i].size() - 1)
                    validWords.push_back(words[i]);
            }
        }
        return validWords;
    }
};