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