【leetcode09-----键盘行】

键盘行

给你一个字符串数组 words ，只返回可以使用在 美式键盘 同一行的字母打印出来的单词。键盘如下图所示。

美式键盘 中：

第一行由字符 "qwertyuiop" 组成。
第二行由字符 "asdfghjkl" 组成。
第三行由字符 "zxcvbnm" 组成。

示例 1：

输入：words = ["Hello","Alaska","Dad","Peace"]
输出：["Alaska","Dad"]
示例 2：

输入：words = ["omk"]
输出：[]
示例 3：

输入：words = ["adsdf","sfd"]
输出：["adsdf","sfd"]

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/keyboard-row
 

题解：

美式键盘从上到下有三行，分别编号0，1，2，定义一个字符串存放26个英文字母所在的行编号，遍历单词数组，每遍历一个单词时，定义一个变量存放该单词第一个字母所在的行编号，然后遍历该单词后面的字母，若后面的字母与第一个字母不在一行，则跳出循环继续遍历下一个单词，否则将该单词加入集合中。最后，将集合中的数组复制到数组中并返回。代码如下：

class Solution{
    public String[] findWords(String[] words){

        //创建一个集合
        List<String> list=new ArrayList<String>();

        //26个字母的行编号字符串
        String rowIdx="12210111011122000010020202";

        //遍历单词数组
        for(String word:words){

            //定义flag初始值为true
            boolean flag=true;

            //Character.toLowerCase()用来将字母转化成小写
            //定义变量存储第一个字母的行编号
            char idx=rowIdx.charAt(Character.toLowerCase(word.charAt(0))-'a');

            //遍历后续字母，判断是否与第一个字母在同一行
            for(int i=1;i<word.length();i++){
                if(rowIdx.charAt(Character.toLowerCase(word.charAt(i))-'a')!=idx){
                    flag=false;
                    break;
                }
            }
            if(flag){
                list.add(word);
            }
        }

        //将集合中的元素复制到数组中并返回
        String[] ans=new String[list.size()];
        for(int i=0;i<list.size();i++){
            ans[i]=list.get(i);
        }
        return ans;
    }
}