500. Keyboard Row

题意就是判断一个单词是否在同一行。

我的方法是：首先判断这个单词的第一个字母在哪个位置，然后判断后边的字母是不是和第一个字母的位置相同，不同的话就赋值0。可能我写的比较麻烦，因为看上去代码好多，有待于进一步改进把。

java代码如下：（运行时间3ms）

 public String[] findWords(String[] words) {
        String line = "QWERTYUIOPASDFGHJKLZXCVBNM";//顺序是按照第一行/第二行/第三行字母来的
        int length = words.length;
        int[] judge = new int[length];//用于判断单词在哪行
        for(int i=0;i<length;i++){
            String s = words[i];
            s = s.toUpperCase();//将单词都变成大写

            //判断第一个字母的位置
            char s1 = s.charAt(0);
            int index = line.indexOf(s1);
            if(index<10)
                judge[i]= 1;
            else if(index<19)
                judge[i] = 2;
            else
                judge[i] = 3;

           //判断其他字母位置与第一个字母位置是否相同，不同就把judge[i]赋值为0；
            for(int j = 1;j<s.length();j++){
                char sj = s.charAt(j);
                int indexj = line.indexOf(sj);
                int j_judge = 0;
                if(indexj<10)
                    j_judge = 1;
                else if(indexj<19)
                    j_judge = 2;
                else
                    j_judge = 3;
                if(j_judge!=judge[i])
                    judge[i] = 0;


            }


        }

       //判断在同一行的单词的个数，用于下一步新建一个新数组初始化
        int sum = 0;
        for(int i=0;i<length;i++){
            if(judge[i]!=0)
                sum++;
        }
        String[] new_words = new String[sum];
        int new_i = 0;

        //将同一行的单词提出来赋值给new_words数组，即可返回
        for(int i=0;i<length;i++){
            if(judge[i]!=0){
                new_words[new_i]=words[i];
                new_i++;
            }
        }
        return new_words;
    }