Go 自定义排序方法

前言

我们经常需要对数组进行排序，Go内置的sort包提供了一些排序函数来帮助我们实现排序。

对于[]int、[]float、[]string这些基础类型数组，使用sort包提供的下面几个函数进行排序：

• sort.Ints
• sort.Float64s
• sort.Strings

看几个例子：

score := []int{5, 4, 3, 2, 1}
sort.Ints(score)
fmt.Println(score) // 输出[1 2 3 4 5]

score := []float64{5.3, 7.1, 5, 1.2}
sort.Float64s(score)
fmt.Println(score) // 输出[1.2 5 5.3 7.1]

s := []string{"kelvin", "allen", "stephen", "leo"}
sort.Strings(s)
fmt.Println(s) // 输出[allen kelvin leo stephen]

可以看到，虽然内置的Ints、Float64s、Strings方法简单好用，但结果总是从小到大的排序。

如果我们想要从大到小，或者想自定义排序规则怎么办？

使用自定义比较器排序

打开sort包，能看到两个自定义比较器：

• sort.Slice
• sort.SliceStable

通过名字我们也能看出来，相较于sort.Slice，sort.SliceStable在排序是会保留相等元素的原始顺序。

比如，如果我想从大到小排序数组[1,2,3,4,5]，可以这样写：

score := []int{1, 2, 3, 4, 5}
sort.Slice(score, func(i, j int) bool {
  return score[i] > score[j]
})
fmt.Println(score) // 输出[5 4 3 2 1]

面对复杂一点的场景，比如对结构体数组排序，又想保留相等元素的原始顺序，可以这样：

user := []struct {
  Name string
  Age  int
}{
  {"kelvin", 20},
  {"allen", 24},
  {"stephen", 24},
  {"leo", 24},
}

sort.SliceStable(user, func(i, j int) bool {
  return user[i].Age > user[j].Age
})
fmt.Println(user) // 输出[{allen 24} {stephen 24} {leo 24} {kelvin 20}]

根据年龄从大到小排列，allen、stephen、leo的原始顺序没变。

总结

本文简述了基本数据类型数组的排序方法，并由此排序方法引出如何自定义比较器进行排序。