当前位置：首页 > article >正文

力扣之字母异位词分组（python）

article 2025/3/13 4:22:04

题目

给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

字母异位词 是由重新排列源单词的所有字母得到的一个新单词。

示例 1:

输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出: [["bat"],["nat","tan"],["ate","eat","tea"]]

示例 2:

输入: strs = [""]
输出: [[""]]

示例 3:

输入: strs = ["a"]
输出: [["a"]]

提示：

1 <= strs.length <= 104
0 <= strs[i].length <= 100
strs[i] 仅包含小写字母

解题思路

取出strs中的每一个单词并用sorted排列，若排列后的值不存在于字典中，则直接将值以列表放入以排列之后的值的valuse中，若以存在，则直接加入到对应的valuse中，最后返回所有的values

代码

class Solution(object):
    def groupAnagrams(self, strs):
        """
        :type strs: List[str]
        :rtype: List[List[str]]
        """
        n = len(strs)
        strs_dict = {}
        for i in range(n):
            new_num = str(sorted(strs[i]))
            print(new_num)
            if new_num in strs_dict:
                strs_dict[new_num].append(strs[i])
            else:
                strs_dict[new_num] = [strs[i]]
        return list(strs_dict.values())