给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。

示例:

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

说明:

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

排序

由于互为字母异位词的两个字符串包含的字母相同,因此对两个字符串分别进行排序之后得到的字符串一定是相同的,故可以将排序之后的字符串作为哈希表的键。

Python

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

复杂度分析

  • 时间复杂度:O(nklog⁡k)O(nk \log k)O(nklogk),其中 nnnstrs\textit{strs}strs 中的字符串的数量,kkkstrs\textit{strs}strs 中的字符串的的最大长度。需要遍历 nnn 个字符串,对于每个字符串,需要 O(klog⁡k)O(k \log k)O(klogk) 的时间进行排序以及 O(1)O(1)O(1) 的时间更新哈希表,因此总时间复杂度是 O(nklog⁡k)O(nk \log k)O(nklogk)

  • 空间复杂度:O(nk)O(nk)O(nk),其中 nnnstrs\textit{strs}strs 中的字符串的数量,kkkstrs\textit{strs}strs 中的字符串的的最大长度。需要用哈希表存储全部字符串。