go语言中sort包支持自定义排序,1. 使用sort.Slice配合比较函数可灵活排序,如按结构体字段升序或降序;2. 实现sort.interface接口(len、less、Swap)适用于复杂或复用场景,可定义ByAge、ByName等类型;3. 多条件排序需在Less或比较函数中组合逻辑,先按主字段排序,相等时按次字段排序,注意处理相等情况。日常推荐sort.Slice,简洁高效,接口方式适合封装。
在Go语言中,sort 包提供了对切片和自定义数据结构进行排序的功能。除了对基本类型(如 int、String)排序外,我们经常需要根据特定规则对结构体或复杂类型排序,这就需要用到自定义排序函数。
1. 使用 sort.Slice 进行自定义排序
Go 1.8 之后引入了 sort.Slice,它允许你传入一个切片和一个比较函数,实现灵活排序。
语法格式:
sort.Slice(slice, func(i, j int) bool {
return 条件表达式
})
示例:对结构体按某个字段排序
立即学习“go语言免费学习笔记(深入)”;
type Person Struct {
Name string
Age int
}
people := []Person{
{“Alice”, 30},
{“Bob”, 25},
{“John”, 35},
}
// 按年龄升序排序
sort.Slice(people, func(i, j int) bool {
return people[i].Age < people[j].Age
})
如果要按姓名降序,可以写成:
sort.Slice(people, func(i, j int) bool {
return people[i].Name > people[j].Name
})
2. 实现 sort.Interface 接口(适用于复杂排序)
对于更复杂的排序逻辑,或希望复用排序规则,可以为类型实现 sort.Interface 接口,它包含三个方法:
- Len() int
- Less(i, j int) bool
- Swap(i, j int)
示例:
type ByAge []Person
func (a ByAge) Len() int { return len(a) }
func (a ByAge) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a ByAge) Less(i, j int) bool { return a[i].Age < a[j].Age }
// 使用
sort.Sort(ByAge(people))
这种方式适合定义多种排序方式,比如再定义一个按名字排序:
type ByName []Person
func (a ByName) Len() int { return len(a) }
func (a ByName) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a ByName) Less(i, j int) bool { return a[i].Name < a[j].Name }
3. 多条件排序(复合排序)
有时需要先按一个字段排序,再按另一个字段排序。可以在 Less 函数中组合条件。
例如:先按年龄升序,年龄相同则按名字升序
sort.Slice(people, func(i, j int) bool {
if people[i].Age == people[j].Age {
return people[i].Name < people[j].Name
}
return people[i].Age < people[j].Age
})
使用 sort.Interface 时逻辑相同,只需在 Less 方法中写复合判断。
基本上就这些。日常开发中 sort.Slice 更简洁常用,而实现接口适合需要封装和复用的场景。多条件排序注意逻辑顺序,避免遗漏相等情况的处理。不复杂但容易忽略细节。