Leetcode 500.键盘行

键盘行

给定一个单词列表，只返回可以使用在键盘同一行的字母打印出来的单词。键盘如下图所示。

   

   

示例：

输入: ["Hello", "Alaska", "Dad", "Peace"]

输出: ["Alaska", "Dad"]

   

注意：

1. 你可以重复使用键盘上同一字符。
2. 你可以假设输入的字符串将只包含字母。

 

import java.util.ArrayList;
import java.util.List;

class Solution {
    public String[] findWords(String[] words) {
        List<String> res=new ArrayList<String>();
        String row1="wqertyuiopQWERTYUIOP";
        String row2="sadfghjklASDFGHJKL";
        String row3="zxcvbnmZXCVBNM";
        for(String word:words){
            int a=0;
            int b=0;
            int c=0;
            for(int i=0;i<word.length();i++){
                if(row1.contains(word.charAt(i)+"")) a=1;
                else if(row2.contains(word.charAt(i)+"")) b=1;
                else if(row3.contains(word.charAt(i)+"")) c=1;
                if(a+b+c>1) break;
            }
            if(a+b+c==1) res.add(word);
        }
        String[] str=new String[res.size()];
        for(int i=0;i<str.length;i++){
            str[i]=res.get(i);
        }
        return str;
    }
}