LeetCode 49.字母异位词分组

题目描述

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

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

示例 1:

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

示例 2:

输入: strs = [""]
输出: [[""]]

示例 3:

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

提示：

• 1 <= strs.length <= 104
• 0 <= strs[i].length <= 100
• strs[i] 仅包含小写字母

题解

方法一：排序

时空复杂度: O(nklogk) O(nk) 其中的n是字符串数组长度，k是字符串的最大长度

这道题目跟上一题十分相似啊LeetCode 242.有效的字母异位词，我们可以在遍历字符串数组的时候对每个字符串进行排序，这样得到的字符串一定是相等的。然后将相同的字符串加入到List中即可。

public List<List<String>> sortMethod(String[] strs) {
    //创建结果集
    Map<String,List<String>> map = new HashMap();
    //遍历字符串数组
    for(String str : strs) {
        //直接sort排序
        char[] chs = str.toCharArray();
        Arrays.sort(chs);
        String s = new String(chs);
        //判断在map中是否出现
        List<String> list = map.getOrDefault(s,new ArrayList<String>());
        //将当前的str存入list中
        list.add(str);
        //存入map中
        map.put(s,list);
    }
    //这里使用new ArrayList(Collection<E> collection)直接创建一个List,因为map.values()返回的是一个Collection
    return new ArrayList(map.values());
}

方法二：哈希表

时空复杂度: O(n(k + 26)) O(n(k + 26)) 其中的n是字符串数组长度，k是字符串的最大长度

我们可以直接遍历每个字符串，然后将字符串中的字符出现的个数拼接后当作key存入map中，这样我们就可以根据固定的key来存储源字符串了

public List<List<String>> hashMethod(String[] strs) {
    //创建结果集
    Map<String,List<String>> map = new HashMap();
    //遍历字符数组
    for(String str : strs) {
        //创建计数数组
        int[] counts = new int[26];
        //遍历字符串计数
        char[] chs = str.toCharArray();
        for(int i = 0; i < chs.length; i ++) {
            counts[chs[i] - 'a'] ++;
        }
        //创建StringBuffer
        StringBuffer sb = new StringBuffer();
        for(int i = 0; i < 26; i ++) {
            //拼接key
            sb.append((char)('a' + i));
            sb.append(counts[i]);
        }
        //生成key
        String key = sb.toString();
        //存入map
        List<String> list = map.getOrDefault(key,new ArrayList());
        list.add(str);
        map.put(key,list);
    }
    return new ArrayList(map.values());
}

总结

其实这道题目的两种方法，底层思想都是类似的，利用哈希表来存储key，然后将不同的异位词存入哈希表中对应的List中，最后直接将map对应的values转换成List返回即可。我们需要读懂题意，搞清楚什么是异位词。