Given an array of strings, group anagrams together.
Example:
Input: ["eat", "tea", "tan", "ate", "nat", "bat"]
Output: [ ["ate","eat","tea"], ["nat","tan"], ["bat"] ]
Note:
- All inputs will be in lowercase.
- The order of your output does not matter
思路
这道题在一开始看到的时候，没有什么具体的思路。但是在想了一会之后觉得可以使用辅助字段来解决该问题。意思就是我们对列表中的字符串进行排序，然后将相同的添加进同一列表中。当遍历完毕之后得到最后的结果。时间复杂度为O(n mlog m),其中n为列表的长度，m为其中最长的字符串长度。空间复杂度为O(n*m)。
在后面看到别人的解答之后学习到了另一种思路就是因为字符只有26个，所有我们设置一个列表，其中包含26个0，然后对列表中每一个字符串进行统计相应的字母出现的次数，然后把他加到相应的位置，然后将其转化为元祖并利用字典，将其对应的字符串添加到对应的列表中。时间复杂度复杂度为O(n*m), n为列表的长度，m为最长的单词数。空间复杂度为O(n*m)。
第一种思路的解决代码
1 class Solution(object):
2 def groupAnagrams(self, strs):
3 ans = collections.defaultdict(list) # 辅助字典
4 for s in strs:
5 ans[tuple(sorted(s))].append(s) # 排序之后将其添加到对应的列表中
6 return ans.values() # 返回列表
第二种思路的解决代码
1 class Solution(object):
2 def groupAnagrams(self, strs):
3 hmap = collections.defaultdict(list)
4 for st in strs:
5 array = [0] * 26 # 26个元素的列表
6 for l in st: # 将字符串元素计数
7 array[ord(l) - ord('a')] += 1
8 hmap[tuple(array)].append(st) # 按照元素出现的个数进行添加
9 return hmap.values()