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

go语言sort包支持自定义排序,1. 使用sort.Slice配合比较函数可灵活排序,如按结构体字段升序或降序;2. 实现sort.interface接口lenless、Swap)适用于复杂或复用场景,可定义ByAge、ByName等类型;3. 多条件排序需在Less或比较函数中组合逻辑,先按主字段排序,相等时按次字段排序,注意处理相等情况。日常推荐sort.Slice,简洁高效,接口方式适合封装

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

在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 更简洁常用,而实现接口适合需要封装和复用的场景。多条件排序注意逻辑顺序,避免遗漏相等情况的处理。不复杂但容易忽略细节。

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