LeetCode:49. Group Anagrams

LeetCode:49. Group Anagrams

Given an array of strings, group anagrams together.

Example:

Input: ["eat", "tea", "tan", "ate", "nat", "bat"],
Output:
[
  ["ate","eat","tea"],
  ["nat","tan"],
  ["bat"]
]

就是找出同一组字符组成的所有可能字符串，并将他们分组。

思路一：对每个字符串排序

先对每个字符串排序，如果排序后的字符串相同，则表明它们是一组。

Python 代码实现

class Solution:
    def sortStr(self, content):
        arr = sorted(content)
        res = ""
        for c in arr:
            res+=c
        return res
        
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        l = len(strs)
        ansMap = {}
        for i in range(l):
            tmp = self.sortStr(strs[i])
            if tmp in ansMap:
                ansMap[tmp].append(strs[i])
            else:
                ansMap[tmp] = [strs[i]]
        ans = []
        for key,value in ansMap.items():
            ans.append(value)
        return ans

时间复杂度

O(NKlogK), N是输入字符串数组的长度，K是其中最长的字符串长度，对这个字符串排序的时间是O(K log K)，所以最终的时间复杂度是O(NK log K)。

思路二：计算字符个数

相当于用字符的个数对某个字符串编码，例如abbccc可以编码为(1,2,3,0,0,…,0)，表示有1个a，2个b，3个c，后面的字母个数都为0

Python 代码实现

class Solution:
    
    def groupAnagrams(self,strs):
        ansDict = collections.defaultdict(list)
        for s in strs:
            count = [0] * 26
            for c in s:
                count[ord(c) - ord('a')] += 1
            ansDict[tuple(count)].append(s)

        return list(ansDict.values())

时间复杂度

由于这里不用排序，只要遍历每个字符，所以里面的复杂度是O(K)，最终结果是O(NK)。

---

THE END.