JAVA代码—算法基础：键盘行字符处理问题

键盘行字符处理问题

问题链接：https://leetcode.com/problems/keyboard-row/description/

问题描述

给定一个单词列表，判断其中哪些单词所包含的字符可以使用下面图例中的键盘上的同一行字母就可以构成。
输出这样的单词。

注意：假设键盘上同一行的字符可以多次使用。
输入的单词列表中所包含的字符仅仅是字母构成。

算法设计

package com.bean.algorithmbasic;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

public class KeyboardRow {

    public static String[] findWords(String[] words) {
        //把键盘上的每一行字母作为一个字符串数组的元素表示出来。（只处理26个字母）
        String[] strs = {"QWERTYUIOP","ASDFGHJKL","ZXCVBNM"};
        //创建一个map对象
        Map<Character, Integer> map = new HashMap<>();
        //将字符串数组元素转化为字符数组，然后依次装入map对象中。
        //map对象的KEY对应字符，VALUE对应行标（rowIndex）
        for(int i = 0; i<strs.length; i++){
            for(char c: strs[i].toCharArray()){
                map.put(c, i);//put <char, rowIndex> pair into the map
            }
        }
        List<String> res = new LinkedList<>();
        for(String w: words){
            if(w.equals("")) continue;
            int index = map.get(w.toUpperCase().charAt(0));
            for(char c: w.toUpperCase().toCharArray()){
                if(map.get(c)!=index){
                    index = -1; //don't need a boolean flag. 
                    break;
                }
            }
            if(index!=-1) res.add(w);//if index != -1, this is a valid string
        }
        return res.toArray(new String[0]);
    }

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        String[] array= {"Hello", "Alaska", "Dad", "Peace"};
        String[] result=findWords(array);
        for(int i=0;i<result.length;i++) {
            System.out.println(result[i]);
        }

    }

}

输出结果：

Alaska
Dad

（完）