LeetCode groupAnagrams Python3

最新推荐文章于 2024-11-22 11:30:20 发布

原创最新推荐文章于 2024-11-22 11:30:20 发布 · 155 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一种有效的算法来解决字母异位词分组问题。通过使用Python中的字典和元组特性，该算法能高效地将一组字符串按字母异位词进行分类。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

示例：
输入： [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出：
[
[“ate”,“eat”,“tea”],
[“nat”,“tan”],
[“bat”]
]

这道题让我知道了，原来在 Python 中 tuple 是 hashable 的，也就是说可以拿一个 tuple 作为字典的 key，但是 list 是不可以的

def groupAnagrams(strs):
	dic = {}
	for word in strs:
		count = [0] * 26
		for l in word:
			pos = ord(l) - ord('a')
			count[pos] += 1
		count = tuple(count)
		if dic.get(count):
			dic[count].append(word)
		else:
			dic[count] = [word]
	return list(dic.values())