本文已参与「新人创作礼」活动, 一起开启掘金创作之路。
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
输入: ["eat", "tea", "tan", "ate", "nat", "bat"]
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
说明:所有输入均为小写字母。
不考虑答案输出的顺序。
思路:
由于互为字母异位词的两个字符串包含的字母相同,因此对两个字符串分别进行排序之后得到的字符串一定是相同的,故可以将排序之后的字符串作为哈希表(go中为map)的键。
时间复杂度:
O(nklogk),其中 n 是 strs 中的字符串的数量,k 是 strs 中的字符串的的最大长度。需要遍历 n 个字符串,对于每个字符串,需要 O(klogk) 的时间进行排序以及 O(1) 的时间更新哈希表,因此总时间复杂度是 O(nklogk)
空间复杂度:
O(nk),其中 nn 是 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)
for _, v := range m {
arr = append(arr, v)
}
return arr
}