详解Go语言如何使用标准库sort对切片进行排序

Ula ·

更新时间:2024-09-20

· 1905 次阅读

sort 标准库

Ints 和 IntsAreSorted

Float64s 和 Float64sAreSorted

Strings 和 StringsAreSorted

Sort

Interface

对结构体切片进行排序

小结

sort 标准库

Sort 标准库提供了对基本数据类型的切片和自定义类型的切片进行排序的函数，常用函数如下表所示：

函数描述

Ints(x []int)	对 int 类型的切片进行升序排序
IntsAreSorted(x []int) bool	判断 int 类型的切片是否是升序排序
Float64s(x []float64)	对 float64 类型的切片进行升序排序
Float64sAreSorted(x []float64) bool	判断 float64 类型的切片是否是升序排序
Strings(x []string)	对 string 类型的切片进行升序排序
StringsAreSorted(x []string) bool	判断 string 类型的切片是否是升序排序
Sort(data Interface)	对自定义类型的切片进行排序，自定义类型必须实现 Interface 接口

如果想了解更多函数的介绍和使用，可以到 pkg.go.dev/sort 进行查看。

Ints 和 IntsAreSorted

Ints(x []int)：对 int 类型的切片进行排序，将切片作为参数进行传递，改变原切片的元素顺序

IntsAreSorted(x []int) bool，传递一个切片进去，判断此切片是否是升序排序，是则返回 true，否则返回 false。

import (
    "fmt"
    "sort"
)
func main() {
    nums := []int{5, 3, 1, 2, 4, 7}
    fmt.Println("排序前：", nums)
    fmt.Println("切片是否是升序排序：", sort.IntsAreSorted(nums))
    sort.Ints(nums)
    fmt.Println("排序后：", nums)
    fmt.Println("切片是否是升序排序：", sort.IntsAreSorted(nums))
}

执行结果：

排序前： [5 3 1 2 4 7]
切片是否是升序排序： false
排序后： [1 2 3 4 5 7]
切片是否是升序排序： true

Float64s 和 Float64sAreSorted

Float64s(x []float64)：对 float64 类型的切片进行排序，将切片作为参数进行传递，改变原切片的元素顺序

Float64sAreSorted(x []float64) bool，传递一个切片进去，判断此切片是否是升序排序，是则返回 true，否则返回 false。

import (
    "fmt"
    "sort"
)
func main() {
    nums := []float64{0.0, 1.5, 1.0, 2.4, 4.4, 7.3}
    fmt.Println("排序前：", nums)
    fmt.Println("切片是否是升序排序：", sort.Float64sAreSorted(nums))
    sort.Float64s(nums)
    fmt.Println("排序后：", nums)
    fmt.Println("切片是否是升序排序：", sort.Float64sAreSorted(nums))
}

执行结果：

排序前： [0 1.5 1 2.4 4.4 7.3]
切片是否是升序排序： false
排序后： [0 1 1.5 2.4 4.4 7.3]
切片是否是升序排序： true

Strings 和 StringsAreSorted

Strings(x []string)：对 float64 类型的切片进行排序，将切片作为参数进行传递，改变原切片的元素顺序

StringsAreSorted(x []string) bool，传递一个切片进去，判断此切片是否是升序排序，是则返回 true，否则返回 false。

import (
    "fmt"
    "sort"
)
func main() {
    strs := []string{"d", "a", "c", "b"}
    fmt.Println("排序前：", strs)
    fmt.Println("切片是否是升序排序：", sort.StringsAreSorted(strs))
    sort.Strings(strs)
    fmt.Println("排序后：", strs)
    fmt.Println("切片是否是升序排序：", sort.StringsAreSorted(strs))
}

执行结果：

排序前： [d a c b]
切片是否是升序排序： false
排序后： [a b c d]
切片是否是升序排序： true

Sort

Sort(data Interface)：对实现 Interface 接口的自定义集合类型按照自定义的规则进行排序。

Interface

实现 Interface 接口，需要实现以下三个方法：

Len() int 返回集合的长度

Less(i, j) bool

i 和 j 为两个元素在集合里的下标值

该方法返回一个 bool 值，判断 i 位置的元素是否应该排在 j 之前，如果 bool 值为 false，则说明 i 不应该在 j 之前。

Swap(i, j int)

对结构体切片进行排序

import (
    "fmt"
    "sort"
)
type User struct {
    Name string
    Age  int
}
type UserSlice []User
func (us UserSlice) Len() int {
    return len(us)
}
func (us UserSlice) Less(i, j int) bool {
    return us[i].Age < us[j].Age
}
func (us UserSlice) Swap(i, j int) {
    us[i], us[j] = us[j], us[i]
}
func main() {
    us := UserSlice{
        User{Name: "小明", Age: 22},
        User{Name: "小红", Age: 17},
        User{Name: "小花", Age: 18},
    }
    fmt.Println("排序前：", us)
    sort.Sort(us)
    fmt.Println("排序后：", us)
}

执行结果：