【LeetCode 500】关关的刷题日记27 Keyboard Row- 专知

【LeetCode 500】关关的刷题日记27 Keyboard Row

点击上方“专知”关注获取更多AI知识!

关关的刷题日记27 – Leetcode 500. Keyboard Row

题目

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.

Example 1: Input: ["Hello", "Alaska", "Dad", "Peace"] Output: ["Alaska", "Dad"] Note: You may use one character in the keyboard more than once. You may assume the input string will only contain letters of alphabet.

题目的意思是给一串单词，让我们找出构成此单词的每一个字母在键盘的同一排的所有单词。键盘上的单词可以被多次使用，每个单词只包含字母。

思路

思路：键盘上一共有三排字母，我们可以建立三个set，然后拿到每个单词的各个字母去这三个set中找，如果此单词的所有字母都在同一个set中，就说明这个单词满足条件。

lass Solution {public:
    vector<string> findWords(vector<string>& words) {
        set<char>row1={'q','w','e','r','t','y','u','i','o','p'};
        set<char>row2={'a','s','d','f','g','h','j','k','l'};
        set<char>row3={'z','x','c','v','b','n','m'};
        vector<string>output;
        int i, j, k,p;
        for( i=0; i<words.size(); i++)
        {
            for( j=0; j<words[i].size(); j++)
            {
                if(row1.find(tolower(words[i][j]))==row1.end())
                    break;
            }
             for( k=0; k<words[i].size(); k++)
            {
                if(row2.find(tolower(words[i][k]))==row2.end())
                    break;
            }
             for( p=0; p<words[i].size(); p++)
            {
                if(row3.find(tolower(words[i][p]))==row3.end())
                    break;
            }
            int n=words[i].size();
            if(j==n || k==n || p==n )
                output.push_back(words[i]);
        }
        return output;
    }};