键盘行

/*
题目思路：
1. 使用Stream流处理，将每个单词映射为一个布尔值，表示它是否可以由键盘同一行字符输入。
2. 过滤出可以由同一行字符输入的单词，并转换为数组。
*/
 
import java.util.Arrays;
import java.util.Set;
import java.util.stream.Collectors;
 
public class Solution {
    public String[] findWords(String[] words) {
        // 定义键盘行集合
        Set<Character> row1 = "qwertyuiopQWERTYUIOP".chars().mapToObj(c -> (char)c).collect(Collectors.toSet());
        Set<Character> row2 = "asdfghjklASDFGHJKL".chars().mapToObj(c -> (char)c).collect(Collectors.toSet());
        Set<Character> row3 = "zxcvbnmZXCVBNM".chars().mapToObj(c -> (char)c).collect(Collectors.toSet());
        
        return Arrays.stream(words)
                .filter(word -> {
                    if (word.isEmpty()) return false;
                    Set<Character> targetRow;
                    if (row1.contains(word.charAt(0))) {
                        targetRow = row1;
                    } else if (row2.contains(word.charAt(0))) {
                        targetRow = row2;
                    } else {
                        targetRow = row3;
                    }
                    return word.chars().allMatch(c -> targetRow.contains((char)c));
                })
                .toArray(String[]::new);
    }
}