14 - Medium - 字谜分组

给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
说明:

所有输入均为小写字母。
不考虑答案输出的顺序。

关键:字典,key为排序后的字母序,value为return_list的下标

class Solution:
    def groupAnagrams(self, strs):
        """
        :type strs: List[str]
        :rtype: List[List[str]]
        """
        d = {}
        i = 0
        res = []
        for _str in strs:
            temp = "".join(sorted(_str))
            if temp in d:
                res[d[temp]].append(_str)
            else:
                d[temp] = i
                res.append([_str])
                i += 1
        return res

关键:字典

class Solution:
    def groupAnagrams(self, strs):
        """
        :type strs: List[str]
        :rtype: List[List[str]]
        """
        d = {}
        for x in strs:
            key = tuple(sorted(x))
            d[key] = d.get(key,[]) + [x]
        return list(d.values())

你可能感兴趣的:(14 - Medium - 字谜分组)