题目
给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
示例 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())