LeetCode 49. 字母异位词分组

LeetCode 49. 字母异位词分组

 

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

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

示例 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] 仅包含小写字母

思路：
由于互为字母异位词的两个字符串包含的字母相同，因此对两个字符串分别进行排序之后得到的字符串一定是相同的，故可以将排序之后的字符串作为哈希表（go中为map）的键。

时间复杂度：
O(nklogk)，其中 n 是 strs 中的字符串的数量，k 是 strs 中的字符串的的最大长度。需要遍历 n 个字符串，对于每个字符串，需要 O(klogk) 的时间进行排序以及 O(1) 的时间更新哈希表，因此总时间复杂度是 O(nklogk)

空间复杂度：
O(nk)，其中 n 是 strs 中的字符串的数量，k 是 strs 中的字符串的的最大长度。需要用哈希表存储全部字符串。

func groupAnagrams(strs []string) [][]string {
    // 归类分组：对数组strs中的每个字符串都进行排序，并将排序后的str作为map的key:
	m := make(map[string][]string)
	for _, str := range strs {
		s := []byte(str)                                             // 拆开成字符数组
		sort.Slice(s, func(i, j int) bool { return s[i] < s[j] })    // sort
		sortStr := string(s)                                         // 转回string
		m[sortStr] = append(m[sortStr], str)
	}

	// 重组结果集
	arr := make([][]string, 0, len(m)
	for _, v := range m {
		arr = append(arr, v)
	}

	return arr
}