方法很好想,用哈希表来做。
只要将每个字符串排序后作为key就可以了
class Solution(object):
def groupAnagrams(self, strs):
"""
:type strs: List[str]
:rtype: List[List[str]]
"""
dict = {}
res = []
for a in sorted(strs):
s = str(sorted(a))
dict[s] = dict.get(s,[]) + [a]
return dict.values()
但是在ac以后发现了两个点:
1、
# 没必要对strs使用排序操作,会导致速度慢一些
for a in sorted(strs):
s = str(sorted(a))
2、之前第一次写的时候没用dict.get函数,导致速度慢了将近7倍。虽然逻辑一致,但是python对get的优化使得速度快了很多。但是还没查到原因。
for a in sorted(strs):
s = str(sorted(a))
'''
dict[s] = dict.get(s,[]) + [a]
'''
if s not in dict.keys():
dict[s] = [a]
else:
dict[s].append(a)