LeetCode之Keyboard Row

题目概述：

给定一个单词列表，返回在键盘上只能用一排字母表示的单词。

思路：

对于给定的单词列表中的每一个单词，把这个单词写进一个新的单词列表里，先判断首字母在哪一排并记录下来，再针对这个单词余下的字母进行判断，若有字母和首字母不在同一排，就把这个单词从新的单词列表中删除。

示例：

Input: ["Hello", "Alaska", "Dad", "Peace"]
Output: ["Alaska", "Dad"]

代码：

class Solution {
public:
    vector<string> findWords(vector<string>& words) {
        set<char> str1={'q','w','e','r','t','y','u','i','o','p'};
        set<char> str2={'a','s','d','f','g','h','j','k','l'};
        set<char> str3={'z','x','c','v','b','n','m'};
        vector<set<char>> str {str1,str2,str3};//把三排字母放在一个vector集合里
        vector<string> word=words;
        vector<string> Word;
        int count=0;
        for(int i=0;i<word.size();i++)
        {
            if(word[i][0]>='A'&&word[i][0]<='Z')
            {
                word[i][0]+=32;
            }//先把字母变为小写
            for(int k=0;k<3;k++)
            {
                if(str[k].count(word[i][0])!=0)
                count=k;
            }//记录首字母在哪一排
            Word.push_back(words[i]);//放进新的单词列表
            for(int j=1;j<word[i].size();j++)
            {
                 if(word[i][j]>='A'&&word[i][j]<='Z')
                {
                    word[i][j]+=32;
                }
                if(str[count].count(word[i][j])==0)
                {
                    Word.pop_back();
                    break;
                }
            }//判断余下字母，若和首字母不在一排就删除
        }
        return Word;
    }
};

总结：

STL中vector：count() 算法用于统计等于某个值的对象的个数 。这里用count()算法，若某个字母存在，计算结果不等于0；若计算结果等于0，则不存在这个字母。