Golang sort排序实现 自定义排序函数写法

go语言中sort包支持切片和自定义数据排序。1. sort.Slice通过比较函数实现灵活排序,如按分数降序、姓名升序;2. 实现sort.interface接口lenless、Swap)可复用排序规则,配合sort.Stable保持稳定;3. 注意Less返回逻辑、使用SliceStable保证稳定性、避免修改数据及处理nil指针。两种方法覆盖多数场景。

Golang sort排序实现 自定义排序函数写法

在 Go 语言中,sort 包提供了对切片和自定义数据结构进行排序的强大功能。除了对基本类型(如 int、String)排序外,还可以通过实现自定义排序函数来满足复杂排序需求。

1. 使用 sort.Slice 进行快速自定义排序

对于切片类型,sort.Slice 是最常用且简洁的方式。它接受一个切片和一个比较函数,适用于结构体或基本类型的灵活排序。

示例:按学生分数降序,分数相同按姓名升序:

 type Student struct {     Name string     Score int }  students := []Student{     {"Alice", 85},     {"Bob", 90},     {"Charlie", 85}, }  sort.Slice(students, func(i, j int) bool {     if students[i].Score == students[j].Score {         return students[i].Name < students[j].Name // 姓名升序     }     return students[i].Score > students[j].Score // 分数降序 }) 

2. 实现 sort.Interface 接口进行精细控制

当需要更复杂的排序逻辑或复用排序规则时,可以为类型实现 sort.Interface 接口的三个方法:Len、Less、Swap。

立即学习go语言免费学习笔记(深入)”;

 type ByScoreThenName []Student  func (a ByScoreThenName) Len() int           { return len(a) } func (a ByScoreThenName) Swap(i, j int)      { a[i], a[j] = a[j], a[i] } func (a ByScoreThenName) Less(i, j int) bool {     if a[i].Score == a[j].Score {         return a[i].Name < a[j].Name     }     return a[i].Score > a[j].Score }  // 使用 sort.Sort(ByScoreThenName(students)) 

这种方式适合定义可复用的排序规则,也可配合 sort.Stable 保持相等元素的原始顺序。

3. 常见使用技巧与注意事项

  • 比较函数中,Less 返回 true 表示第 i 个元素应排在第 j 个前面
  • 使用 sort.SliceStable 可保证相等元素的相对顺序不变
  • 避免在 Less 或比较函数中修改数据,可能导致排序错误或崩溃
  • 对于指针切片,注意 nil 值处理,防止 panic

基本上就这些。sort.Slice 适合大多数场景,接口实现适合需要复用或稳定排序的情况。掌握这两种方式,就能灵活应对各种排序需求。不复杂但容易忽略细节。

© 版权声明
THE END
喜欢就支持一下吧
点赞14 分享