1、题目描述
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
输入: ["eat", "tea", "tan", "ate", "nat", "bat"]
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
- 所有输入均为小写字母。
- 不考虑答案输出的顺序。
2、代码详解
先排序,再用字典dict
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
dic = {}
for s in strs:
keys = "".join(sorted(s))
if keys not in dic:
dic[keys] = [s]
else:
dic[keys].append(s)
return list(dic.values())
或者使用
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
hash_map = collections.defaultdict(list)
for s in strs:
k = "".join(sorted(s))
hash_map[k].append(s)
return list(hash_map.values())
- 当使用
collections.defaultdict(list)
创建一个字典时,对于任何不存在的键,访问该键将返回一个空的列表[]
作为默认值。 - 这在需要动态地构建一个以某种类型为值的字典时非常方便,无需手动检查键是否存在并初始化相应的值。
- 当遍历列表
strs
中的每个字符串s
时,通过sorted(s)
对字符串进行排序,然后使用"".join(sorted(s))
将排序后的字符列表转换为字符串作为键k
。如果键k
已经存在于字典中,说明之前已经有相同字母排序结果的字符串,直接将当前字符串s
添加到对应键的值列表中(hash_map[k].append(s)
);如果键k
不存在,由于使用了defaultdict(list)
,会自动创建一个新的空列表作为该键的值,并将当前字符串s
添加进去。collections.defaultdict(list)
在这段代码中简化了字典的操作,避免了手动处理不存在的键的情况,使得代码更加简洁和高效。