给定一个单词列表,只返回可以使用在键盘同一行的字母打印出来的单词。键盘如下图所示。
示例:
输入: ["Hello", "Alaska", "Dad", "Peace"]
输出: ["Alaska", "Dad"]
方法1:键盘上一共有三排字母,我们可以建立三个哈希表,然后拿到每个单词的各个字母去这三个哈希表中找,如果有任意两个字符出现在不同的表中,就说明这个单词不满足条件。
class Solution {
public:
vector<string> findWords(vector<string>& words) {
vector<string> re;
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'};
for(auto x:words){
bool a1=false, a2=false, a3=false;
for(auto y:x){
y=tolower(y);
if(row1.count(y)) a1=true;
if(row2.count(y)) a2=true;
if(row3.count(y)) a3=true;
if(a1+a2+a3>1) break;
}
if(a1+a2+a3==1) re.push_back(x);
}
return re;
}
};
方法2:用到了find_first_of,代码更简洁一些。
class Solution {
public:
vector<string> findWords(vector<string>& words) {
vector<string> re;
for(auto x:words){
bool a1=x.find_first_of("QWERTYUIOPqwertyuiop")==-1 ? false:true;
bool a2=x.find_first_of("ASDFGHJKLasdfghjkl")==-1 ? false:true;
bool a3=x.find_first_of("ZXCVBNMzxcvbnm")==-1 ? false:true;
if(a1+a2+a3==1) re.push_back(x);
}
return re;
}
};