leetcode 49. 字母异位词分组(中等)
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
输入: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出: [
[“ate”,“eat”,“tea”],
[“nat”,“tan”],
[“bat”]
]
步骤
一开始写的代码超时了,看了官方的题解才意识到还有字典配数组的方法,觉得自己是真的蠢。
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
ans=[]
while strs:
temp=[strs[0]]
tmp=sorted(strs[0])
strs.remove(strs[0])
for i in strs[::-1]:
if sorted(i)==tmp:
temp.append(i)
strs.remove(i)
ans.append(temp)
return ans
mp = collections.defaultdict(list),字典中若没有该关键字,则默认将它的值设为一个空数组。
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
mp = collections.defaultdict(list)
for st in strs:
key = "".join(sorted(st))
mp[key].append(st)
return list(mp.values())