Sui Xin's Blog

[LeetCode]49. 字母异位词分组（Group Anagrams）

Posted on In Disqus:

题目描述

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

示例:

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

说明：

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

解题思路

哈希表。

  1. 解法一：字符串排序。将字符串排序后分为不同的类添加到哈希表中。
  2. 解法二：字符计数。因为全部是小写字母，共 26 个，可直接计数来分为不同的类。

注：字典的键必须为不可变类，list 为可变类，故转变为 tuple。

代码

Python 3.6

解法一：字符串排序

1
2
3
4
5
6
class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        res = {}
        for s in strs:
            res.setdefault(tuple(sorted(s)), []).append(s)
        return list(res.values())

执行用时 : 152 ms
内存消耗 : 17.7 MB

解法二：字符计数

1
2
3
4
5
6
7
8
9
class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        res = {}
        for s in strs:
            count = [0] * 26
            for char in s:
                count[ord(char) - ord("a")] += 1
            res.setdefault(tuple(count), []).append(s)
        return list(res.values())

执行用时 : 280 ms
内存消耗 : 18.8 MB

参考

https://leetcode-cn.com/problems/group-anagrams/solution/zi-mu-yi-wei-ci-fen-zu-by-leetcode/

Related Posts