写点什么

Go 语言中排序的 3 种方法

作者:AlwaysBeta
  • 2023-08-18
    北京
  • 本文字数:1303 字

    阅读完需:约 4 分钟

原文链接: Go 语言中排序的 3 种方法


在写代码过程中,排序是经常会遇到的需求,本文会介绍三种常用的方法。


废话不多说,下面正文开始。

使用标准库

根据场景直接使用标准库中的方法,比如:


  • sort.Ints

  • sort.Float64s

  • sort.Strings


举个例子:


s := []int{4, 2, 3, 1}sort.Ints(s)fmt.Println(s) // [1 2 3 4]
复制代码

自定义比较器

使用 sort.Slice 方法排序时,可以自定义比较函数 less(i, j int) bool,这样就可以根据需要按不同的字段进行排序。


如果想要稳定排序的话,就使用 sort.SliceStable 方法。


举个例子:


family := []struct {    Name string    Age  int}{    {"Alice", 23},    {"David", 2},    {"Eve", 2},    {"Bob", 25},}
// Sort by age, keeping original order or equal elements.sort.SliceStable(family, func(i, j int) bool { return family[i].Age < family[j].Age})fmt.Println(family) // [{David 2} {Eve 2} {Alice 23} {Bob 25}]
复制代码

自定义数据结构

使用 sort.Sort 或者 sort.Stable 方法,它们可以对任意实现了 sort.Interface 的数据结构排序。


type Interface interface {    // Len is the number of elements in the collection.    Len() int    // Less reports whether the element with    // index i should sort before the element with index j.    Less(i, j int) bool    // Swap swaps the elements with indexes i and j.    Swap(i, j int)}
复制代码


意思就是说,只要某一个数据结构实现了 Len() intLess(i, j int) boolSwap(i, j int) 这三个方法,那么就可以使用 sort.Sort 来排序。


举个例子:


type Person struct {    Name string    Age  int}
// ByAge implements sort.Interface based on the Age field.type ByAge []Person
func (a ByAge) Len() int { return len(a) }func (a ByAge) Less(i, j int) bool { return a[i].Age < a[j].Age }func (a ByAge) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func main() { family := []Person{ {"Alice", 23}, {"Eve", 2}, {"Bob", 25}, } sort.Sort(ByAge(family)) fmt.Println(family) // [{Eve 2} {Alice 23} {Bob 25}]}
复制代码

字典排序

我们都知道,字典是无序的,具体原因可以看之前写的这篇文章 Go 语言 map 如何顺序读取?


如果想要字典按 key 或者 value 排序的话,可以这样做。


m := map[string]int{"Alice": 2, "Cecil": 1, "Bob": 3}
keys := make([]string, 0, len(m))for k := range m { keys = append(keys, k)}sort.Strings(keys)
for _, k := range keys { fmt.Println(k, m[k])}// Output:// Alice 2// Bob 3// Cecil 1
复制代码


以上就是本文的全部内容,如果觉得还不错的话欢迎点赞转发关注,感谢支持。




参考文章:


  • https://yourbasic.org/golang/how-to-sort-in-go/#performance-and-implementation


推荐阅读:


发布于: 刚刚阅读数: 4
用户头像

AlwaysBeta

关注

微信公众号:AlwaysBeta 2017-11-30 加入

专注分享后端开发技术干货!

评论

发布
暂无评论
Go 语言中排序的 3 种方法_Go_AlwaysBeta_InfoQ写作社区