【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] 仅包含小写字母

计数法：

通过计算每个字母的个数来判断两个单词是不是异位词。
因为我没有想到用哈希表来存储值，最后提交后超时没能通过。

所以这里提醒：一旦需要根据特征进行归类，就应该利用散列表。

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        n = len(strs)
        if n == 1:
            return [strs]

        def same(str1, str2):
            if str1 == str2:
                return True
            count1 = {}
            count2 = {}
            for i in str1:
                count1[i] = count1.get(i,0) + 1
            for i in str2:
                count2[i] = count2.get(i,0) + 1
            if  len(count1) == 0 or len(count2)==0 or len(count1) != len(count2):
                return False
            for k, v in count1.items():
                if count2.get(k,0) != v:
                    return False
            return True
        result = []
        while(len(strs) > 0):
            tmp = []
            n = len(strs)
            str1 = strs[0]
            tmp.append(str1)
            if n > 1:
                for i in range(1,n):
                    if same(str1, strs[i]):
                        tmp.append(strs[i])
            result.append(tmp)
            for t in tmp:
                strs.remove(t)
        return result

计数法（用哈希表）：

由于互为字母异位词的两个字符串包含的字母相同，因此两个字符串中的相同字母出现的次数一定是相同的，故可以将每个字母出现的次数使用字符串表示，作为哈希表的键。

由于字符串只包含小写字母，因此对于每个字符串，可以使用长度为 26 的数组记录每个字母出现的次数

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        mp = collections.defaultdict(list)

        for st in strs:
            counts = [0] * 26
            for ch in st:
                counts[ord(ch) - ord("a")] += 1
            # 需要将 list 转换成 tuple 才能进行哈希
            mp[tuple(counts)].append(st)
        
        return list(mp.values())

时间复杂度：$O(Nk)$

排序法：

把一个字符串拆成 char list 然后按字母进行排序。字母异位词 之间排序后的值是相等的。

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        result = collections.defaultdict(list)
        for str in strs:
            chars = ''.join(sorted(str))
            result[chars].append(str)
        return list(result.values())

时间复杂度：$O(Nk\log k)$

参考文献：
[1] 力扣官方题解