LeetCode 49. 字母异位词分组

奋斗中的Jack2024-01-10 19:39

49. 字母异位词分组

给你一个字符串数组，请你将字母异位词组合在一起。可以按任意顺序返回结果列表。

字母异位词是由重新排列源单词的所有字母得到的一个新单词。

示例 1:

输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]

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

示例 2:

输入: strs = [""]

输出:[[""]]

示例 3:

**输入:**strs = ["a"]

输出: [["a"]]

提示：

1 <= strs.length <= 10^4
0 <= strs[i].length <= 100
strs[i] 仅包含小写字母

解法思路：

1、排序

2、计数

法一：

java 复制代码

class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
        // 排序, n 是 strs 中字符串数量，k 是 strs 中字符串最大长度
        // Time: O(nklogk)
        // Space: O(nk)
        Map<String, List<String>> map = new HashMap<String, List<String>>();
        for (String str : strs) {
            char[] chs = str.toCharArray();
            Arrays.sort(chs);
            String key = new String(chs);
            List<String> list = map.getOrDefault(key, new ArrayList<String>());
            list.add(str);
            map.put(key, list);
        }
        return new ArrayList<List<String>>(map.values());
    }
}

法二：

java 复制代码

class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
        // 计数, n = strs.length，k = Max(str.length()), ∣Σ∣ = 26
        // Time: O(n(k+∣Σ∣))
        // Space: O(n(k+∣Σ∣))
        Map<String, List<String>> map = new HashMap<String, List<String>>();
        for (String str : strs) {
            int[] counts = new int[26];
            int len = str.length();
            for (int i = 0; i < len; i++) {
                counts[str.charAt(i) - 'a']++;
            }
            StringBuffer sb = new StringBuffer();
            for (int i = 0; i < 26; i++) {
                if (counts[i] != 0) {
                    sb.append((char) ('a' + i));
                    sb.append(counts[i]);
                }
            }
            String key = sb.toString();
            List<String> list = map.getOrDefault(key, new ArrayList<String>());
            list.add(str);
            map.put(key, list);
        }
        return new ArrayList<List<String>>(map.values());
    }
}