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;
     }
 }