程序员面试金典 - 面试题 10.02. 变位词组

题目难度: 中等

原题链接

今天继续更新程序员面试金典系列, 大家在公众号 算法精选 里回复 面试金典 就能看到该系列当前连载的所有文章了, 记得关注哦~

题目描述

编写一种方法，对字符串数组进行排序，将所有变位词组合在一起。变位词是指字母相同，但排列不同的字符串。

注意：本题相对原题稍作修改

示例:

• 输入: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”],
• 输出:

[
  ["ate","eat","tea"],
  ["nat","tan"],
  ["bat"]
]

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

题目思考

1. 变位词有什么特点?

解决方案

思路

• 根据题目描述, 变位词是指字母相同, 但排列不同的字符串
• 那么如果将这些变位词自身按照字典序排序, 其排序后的字符串一定相同
• 我们可以利用这一特点, 新建一个键为排序后的字符串, 值用于存储原始字符串列表的字典
• 然后我们就可以将各个单词排序后的字符串作为字典的 key, 并将原始字符串存储在字典的值列表中
• 最终结果即为字典的所有值列表
• 下面的代码就对应了上面的整个过程, 并且有详细的注释, 方便大家理解

复杂度

• 时间复杂度 O(NMlogM): N 是字符串数目, M 是字符串长度, 需要遍历所有字符串, 然后内部需要对字符串进行排序, 所以时间复杂度是O(NMlogM)
• 空间复杂度 O(NM): 额外使用了一个字典, 包含所有字符串

代码

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        # 排序后字符串作为key
        d = collections.defaultdict(list)
        for s in strs:
            # key就是排序后的字符串
            key = "".join(sorted(s))
            # 将原始字符串加入字典键key对应的列表中
            d[key].append(s)
        # 最终结果即为字典所有的值, 每个值是一个变位词列表
        return list(d.values())

---

大家可以在下面这些地方找到我~

我的 GitHub

我的 Leetcode

我的 CSDN

我的知乎专栏

我的头条号

我的牛客网博客

我的公众号: 算法精选, 欢迎大家扫码关注~