打败算法 —— 字母异位词分组
本文参考
出自LeetCode上的题库 —— 字母异位词分组,本题考察哈希表的应用
https://leetcode-cn.com/problems/group-anagrams/
字符串解码问题
给你一个字符串数组,请你将字母异位词组合在一起。可以按任意顺序返回结果列表
字母异位词是由重新排列源单词的字母得到的一个新单词,所有源单词中的字母通常恰好只用一次
示例1:
输入:strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出:[["bat"],["nat","tan"],["ate","eat","tea"]]
示例 2:
输入:strs = [""]
输出:[[""]]
示例 3:
输入:strs = ["a"]
输出:[["a"]]
解题思路
本题的关键是如何设计哈希表的键,让异位词映射到同一个键下,因为题目还附加了"只有小写字母"的条件,所以可以构造大小为26的列表,每个索引记录字母出现的次数,并将这个list作为键,实现异位词的映射
哈希表解法
import collections
class Solution:
def group_anagrams(self, strs: List[str]) -> List[List[str]]:
mp = collections.defaultdict(list)
for s in strs:
count = [0] * 26
for c in s:
count[ord(c) - ord('a')] += 1
mp[tuple(count)].append(s)
return list(mp.values())