LeetCode 49. 字母异位词分组 | Python

49. 字母异位词分组

---

题目来源：LeetCode（力扣）https://leetcode-cn.com/problems/group-anagrams/

题目

---

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

示例:

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

说明：

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

解题思路

---

思路：排序、计数

先审题，题目要求在给定的字符串数组中，将字母异位词组合在一起。

字母异位词：

• 这里指字母相同，但排列不同的字符串。

排序

也就是说，如果是 字母异位词，那么相应字符串所包含的字符一定是相同的。那么如果对这些字符串进行排序之后，它们也一定是相同。

此时，我们将排序后的字符串作为哈希表的键，而值则为排序前的字符串，也就是字母异位词组合的列表。

具体的代码实现如下。

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

        # 遍历数组，对字符串进行排序作为键
        for string in strs:
            key = ''.join(sorted(string))
            anagrams[key].append(string)
        
        return [value for value in anagrams.values()]

计数

除了排序的方法，我们这里还可以通过计数来判断是否是 字母异位词。

如果是 字母异位词，字符串之间包含的字符是相同的，那么这些字符串的相应字母出现的次数也一定是相同的。那么在这里将字母出现的次数作为哈希表的键。

题目中提示，所有输入都只包含小写字母。那么这里声明一个长度为 26 的列表，存储每个字符串字符出现的次数。因为键是不可变，这里将列表转换为元组，作为哈希表的键。

具体的代码实现如下。

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

        for string in strs:
            # 声明长度 26 的列表用于计算字符出现的次数
            ch_counts = [0] * 26
            for ch in string:
                # 统计字符串出现次数
                ch_counts[ord(ch)-ord('a')] += 1
            # 将列表转换为元组，作为键，将当前字符串添加到值中
            anagrams[tuple(ch_counts)].append(string)
        
        return [value for value in anagrams.values()]

欢迎关注

---

公众号 【书所集录】

---

如有错误，烦请指出，欢迎指点交流。