Go 语言中排序的三种方法

开发 后端
使用 Sort.Slice​ 方法排序时,可以自定义比较函数 less(i, j int) bool,这样就可以根据需要按不同的字段进行排序。如果想要稳定排序的话,就使用 Sort.SliceStable 方法。

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

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

使用标准库

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

  • 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() int,Less(i, j int) bool 和 Swap(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
责任编辑:姜华 来源: yongxinz
相关推荐

2020-08-12 08:51:19

Go语言Concurrency后台

2021-12-20 07:11:26

Java List排序 Java 基础

2019-09-09 15:13:19

Yum包管理器Linux

2022-05-31 16:00:46

Go 编程语言复制文件Go 标准库

2009-07-08 12:56:32

编写Servlet

2020-03-13 18:10:08

Linuxapt软件包

2011-06-10 10:43:12

Ubuntu应用安装

2009-06-23 10:45:18

Hibernate支持

2010-09-14 15:10:49

CSS注释

2009-12-11 18:49:39

预算编制博科资讯

2011-04-18 15:32:45

游戏测试测试方法软件测试

2022-07-13 16:06:16

Python参数代码

2023-08-14 17:58:13

RequestHTTP请求

2010-09-08 13:29:48

CSS

2010-11-16 16:11:28

Oracle身份验证

2020-06-17 10:52:00

DDoS攻击网络攻击网络安全

2016-10-12 13:53:38

JavaByteBufferRandomAcces

2023-02-21 14:58:12

间序列周期数据集

2023-05-16 16:07:07

大数据数据管理工具

2016-09-09 13:07:56

CentOSJDKLinux
点赞
收藏

51CTO技术栈公众号