leetcode 49. 字母异位词分组（中等）

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())