Java实现 LeetCode 500 键盘行

500. 键盘行

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

示例：

输入: [“Hello”, “Alaska”, “Dad”, “Peace”]

输出: [“Alaska”, “Dad”]

注意：

你可以重复使用键盘上同一字符。

你可以假设输入的字符串将只包含字母。

class Solution {
     public String[] findWords(String[] words)
    {
        String[] rec = new String[3];
        rec[0] = "qwertyuiop";
        rec[1] = "asdfghjkl";
        rec[2] = "zxcvbnm";

        List<String> list = new ArrayList<>();

        for(String str : words)
        {
            for(int j = 0; j < 3; j++)
            {
                boolean flag = true;

                for(int k = 0; k < str.length(); k++)
                {
                    char ch = str.charAt(k);

                    //转换成小写
                    if(ch >= 'A' && ch <= 'Z')
                        ch = (char)(ch+32);

                    //在这一行中找不到这个单词，换下一行
                    if(rec[j].indexOf(ch) == -1)
                    {
                        flag = false;
                        break;
                    }
                }

                if(flag == true)
                {
                    list.add(str);
                    break;
                }
            }
        }
        return list.toArray(new String[]{});

    }
}