【Leetcode】49. 字母异位词分组

题目链接

【Leetcode】49. 字母异位词分组

代码一

func groupAnagrams(strs []string) [][]string {
	// 存放字典序相同的字符串切片
	hash := map[string][]string{}
	// 枚举给定的字符串数组，如果字符串的字典序相同，则放到对应的切片中
	for _, str := range strs {
		// 把字符串转成byte数组
		arr := []byte(str)
		// 用内置库函数对切片进行排序
		sort.Slice(arr, func(i, j int) bool { return arr[i] < arr[j] })
		// 把字节切片转成字符串得到字典序排序后的字符串
		sortedStr := string(arr)
		hash[sortedStr] = append(hash[sortedStr], str)
	}

	//	如果字符串按照字典序排序之后，两个字符串相等，则为字母异位词
	ans := make([][]string, 0, len(hash))
	// 枚举map，把所有切片放到结果中
	for _, arr := range hash {
		ans = append(ans, arr)
	}
	return ans
}

代码二

时间复杂度较高

func groupAnagrams(strs []string) [][]string {
	// 存放字典序相同的字符串切片
	hash := map[string][]string{}
	// 枚举给定的字符串数组，如果字符串的字典序相同，则放到对应的切片中
	for _, str := range strs {
		sortedStr := Sort(str)
		hash[sortedStr] = append(hash[sortedStr], str)
	}

	//	如果字符串按照字典序排序之后，两个字符串相等，则为字母异位词
	ans := [][]string{}
	// 枚举map，把所有切片放到结果中
	for _, arr := range hash {
		ans = append(ans, arr)
	}
	return ans
}

// 对单个字符串进行字典序排序
func Sort(str string) string {
	// 把字符串的每个字符拆成单个字符串放到字符串切片arr
	arr := []string{}
	for _, val := range str {
		arr = append(arr, string(val))
	}
	// 调用sort函数，对字符串切片进行排序
	sort.Strings(arr)
	// 把排好序之后的字符串重新拼接成一个字符串
	sortedStr := ""
	for _, val := range arr {
		sortedStr += val
	}
	// 返回升序排序好的字符串
	return sortedStr
}