leetcode之49字母异位词分组Golang

题目描述

给定一个字符串数组，将字母异位词组合在一起。字母异位词指字母相同，但排列不同的字符串。

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"]
输出:
[
  ["ate","eat","tea"],
  ["nat","tan"],
  ["bat"]
]

说明：

• 所有输入均为小写字母。
• 不考虑答案输出的顺序。

算法

核心思想就是创建一个map，它的键是经过排序后的字符串，它的值就是具有相同字符的字符串数组

• 遍历入口参数数组，获得一个字符串
• 对当前字符串进行排序，将排序后的字符串作为键，原始的字符串作为值加入到map
• 当遍历完整个入口参数数组以后，将整个map的值组合成一个二维字符串数组返回

代码

内置类型不能够给他添加方法，需要先使用type关键字将类型转化为自定义类型才可以添加方法：

type byteSlice []byte

可以使用内置的sort.Sort对字符串排序，但是需要定义以下方法来实现sort.Interface接口

// 返回集合中元素的个数
func (b byteSlice) Len() int {
	return len(b)
}
// 判断索引i的元素的值是否比索引j的元素小
func (b byteSlice) Less(i, j int) bool {
	if b[i] < b[j] {
		return true
	}
	return false
}
// 交换索引i和索引j的元素的位置
func (b byteSlice) Swap(i, j int) {
	b[i], b[j] = b[j], b[i]
}

下面就是算法的代码

func groupAnagrams(strs []string) [][]string {
	resMap := make(map[string][]string)
	for _, word := range strs {
		tmpStr := byteSlice(word)
		sort.Sort(byteSlice(tmpStr))
		resMap[string(tmpStr)] = append(resMap[string(tmpStr)], word)
	}
	resStrings := make([][]string, 0)
	for _, strings := range resMap {
		resStrings = append(resStrings, strings)
	}
	return resStrings
}