49. 字母异位词分组

小豆丁 1年前 ⋅ 1128 阅读
49. 字母异位词分组
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"]
输出:
[
  ["ate","eat","tea"],
  ["nat","tan"],
  ["bat"]
]
说明:

所有输入均为小写字母。
不考虑答案输出的顺序。

解析

我们直接使用字母数量作为key,然后进行分组。这里的技巧就是利用都是小写字母进行数组归纳。

class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
        if(strs==null || strs.length==0) return null;
        Map<String,List<String>> map = new HashMap<>();
        for(String str : strs){
            int[] dict = new int[26];
            for(char val : str.toCharArray()){
                dict[(int)(val-'a')]++;
            }
            String key = Arrays.toString(dict);
            List<String> lst = map.getOrDefault(key,new ArrayList<String>());
            lst.add(str);
            map.put(key,lst);
        }
        return new ArrayList<>(map.values());
    }
}