Go 自定义排序

有的时候我们希望排序不是仅仅按照自然顺序排序。例如，我们希望按照字符串的长度来对一个字符串数组排序而不是按照字母顺序来排序。这里我们介绍一下Go的自定义排序。

  1. package main
  3. import "sort"
  4. import "fmt"
  6. // 为了能够使用自定义函数来排序，我们需要一个
  7. // 对应的排序类型，比如这里我们为内置的字符串
  8. // 数组定义了一个别名ByLength
  9. type ByLength []string
  11. // 我们实现了sort接口的Len，Less和Swap方法
  12. // 这样我们就可以使用sort包的通用方法Sort
  13. // Len和Swap方法的实现在不同的类型之间大致
  14. // 都是相同的，只有Less方法包含了自定义的排序
  15. // 逻辑，这里我们希望以字符串长度升序排序
  16. func (s ByLength) Len() int {
  17.     return len(s)
  18. }
  19. func (s ByLength) Swap(i, j int) {
  20.     s[i], s[j] = s[j], s[i]
  21. }
  22. func (s ByLength) Less(i, j int) bool {
  23.     return len(s[i]) < len(s[j])
  24. }
  26. // 一切就绪之后，我们就可以把需要进行自定义排序
  27. // 的字符串类型fruits转换为ByLength类型，然后使用
  28. // sort包的Sort方法来排序
  29. func main() {
  30.     fruits := []string{"peach", "banana", "kiwi"}
  31.     sort.Sort(ByLength(fruits))
  32.     fmt.Println(fruits)
  33. }

输出结果

  1. [kiwi peach banana]

同样的，对于其他的类型，使用这种方法，我们可以为Go的切片提供任意的排序方法。归纳一下就是：

  1. 创建自定义排序类型
  2. 实现sort包的接口方法Len,Swap和Less
  3. 使用sort.Sort方法来排序