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:

1. You may use one character in the keyboard more than once.
2. You may assume the input string will only contain letters of alphabet. 

找字符串，该字符串的每个字母都要在美式键盘的同一行即可。

 

 

 

class Solution {
public:
    vector<string> findWords(vector<string>& words) {
        vector<string> ans = {"qwertyuiopQWERTYUIOP","asdfghjklASDFGHJKL","zxcvbnmZXCVBNM"};
        vector<string> findans;
        for(auto word = words.begin();word!=words.end();word++){
            string s = *word;
            int n = 0,i = 1;
            if(ans[0].find(s[0]) != s.npos)
                ;
            else if(ans[1].find(s[0]) != s.npos)
                n = 1;
            else
                n = 2;
            for(;i < s.length();i++)
                if(ans[n].find(s[i]) == s.npos)
                    break;
            if(i == s.length())
                findans.push_back(s);
        }
        return findans;
    }
};

 

 python：

class Solution(object):
    def findWords(self, words):
        row1,row2,row3 = set('qwertyuiop'),set('asdfghjkl'),set('zxcvbnm')
        ret = []
        for word in words:
            w = set(word.lower())
            if w.issubset(row1) or w.issubset(row2) or w.issubset(row3):
                ret.append(word)
        return ret

　　

class Solution(object):
    def findWords(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        a=set('qwertyuiop')
        b=set('asdfghjkl')
        c=set('zxcvbnm')
        ans=[]
        for word in words:
            t=set(word.lower())
            if a&t==t:
                ans.append(word)
            if b&t==t:
                ans.append(word)
            if c&t==t:
                ans.append(word)
        return ans