leetcode 49 字母异位分词

正文

基础解法

首先，我们创建一个字典对象，然后遍历整个字符串列表，并且使用 sorted() 函数对字符串列表进行排序，所有的异位分词经过排序后它们的组成和顺序会趋于一致。但是需要注意的是 sorted 对字符串进行排序后会变成一个由单个独立字符组成的列表，我们需要使用 join 将它们重新连接为字符串。进一步我们判断组合后的字符串 tmp 是否位于我们创建的字典对象 dict1 中，如果不在，就将原始位于 strs 中的字符串以值的方式添加进入字典，这里特别需要注意的一点是，一定要以列表类型进行添加。对应下面代码中的 dict1[tmp] = [s]。如果 tmp 在 dict1 中，就是用列表的 append() 方法将字符串添加进入字典中。最后使用 dict1.values() 获取字典对应的值即可。

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        dict1 = {}
        for s in strs:
            tmp = "".join(sorted(s))
            if tmp in dict1:
                dict1[tmp].append(s)
            else:
                dict1[tmp] = [s]
                
        return list(dict1.values())

进阶解法

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        dict1 = collections.defaultdict(list)
        for s in strs:
            tmp = "".join(sorted(s))          
            dict1[tmp].append(s)
                
        return list(dict1.values())

这里我们通过 Python 中的 collections 模块定义默认字典，就可以去掉使用 else 来对 tmp 不在 dict1 中时进行补充的情况。注意要在collections.defaultdict(list) 中指明 list，这样我们才能够在后面使用 append() 命令进行元素添加。

如果大家觉得有用，就请点个赞吧~