博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
golang 对自定义类型排序
阅读量:5733 次
发布时间:2019-06-18

本文共 3754 字,大约阅读时间需要 12 分钟。

golang 对自定义类型排序

在实际项目中,我们常常需要根据一个结构体类型的某个字段进行排序。之前遇到这个问题不知道如何解决,后来在网上搜索了相关问题,找到了一些好的解决方案,此处参考下,做个总结吧。

由于 golang 的 sort 包本身就提供了相应的功能, 我们就没必要重复的造个轮子了,来看看如何利用 sort 包来实现吧。

sort包浅谈

sort 包 在内部实现了四种基本的排序算法:插入排序(insertionSort)、归并排序(symMerge)、堆排序(heapSort)和快速排序(quickSort); sort 包会依据实际数据自动选择最优的排序算法。所以我们写代码时只需要考虑实现 sort.Interface 这个类型就可以了。

粗略的看看sort包

func Sort(data Interface) {    // Switch to heapsort if depth of 2*ceil(lg(n+1)) is reached.    n := data.Len()    maxDepth := 0    for i := n; i > 0; i >>= 1 {        maxDepth++    }    maxDepth *= 2    quickSort(data, 0, n, maxDepth)}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)}// 内部实现的四种排序算法// 插入排序func insertionSort(data Interface, a, b int)// 堆排序func heapSort(data Interface, a, b int)// 快速排序func quickSort(data Interface, a, b, maxDepth int)// 归并排序func symMerge(data Interface, a, m, b int)

所以要调用 sort.Sort() 来实现自定义类型排序,只需要我们的类型实现 Interface 接口类型中的三个方法即可。

先看看 sort 包本身对于 []int 类型如何排序

// 首先定义了一个[]int类型的别名IntSlice type IntSlice []int// 获取此 slice 的长度func (p IntSlice) Len() int           { return len(p) }// 比较两个元素大小 升序func (p IntSlice) Less(i, j int) bool { return p[i] < p[j] }// 交换数据func (p IntSlice) Swap(i, j int)      { p[i], p[j] = p[j], p[i] }// sort.Ints()内部调用Sort() 方法实现排序// 注意 要先将[]int 转换为 IntSlice类型 因为此类型才实现了Interface的三个方法 func Ints(a []int) { Sort(IntSlice(a)) }

照葫芦画瓢 我们来对自定义的结构体类型进行降序排序

package mainimport (    "fmt"    "sort")type Person struct {    Name string    Age  int}type Persons []Person// 获取此 slice 的长度func (p Persons) Len() int { return len(p) }// 根据元素的年龄降序排序 (此处按照自己的业务逻辑写) func (p Persons) Less(i, j int) bool {    return p[i].Age > p[j].Age}// 交换数据func (p Persons) Swap(i, j int) { p[i], p[j] = p[j], p[i] }func main() {    persons := Persons{        {            Name: "test1",            Age:  20,        },        {            Name: "test2",            Age:  22,        },        {            Name: "test3",            Age:  21,        },    }    fmt.Println("排序前")    for _, person := range persons {        fmt.Println(person.Name, ":", person.Age)    }    sort.Sort(persons)    fmt.Println("排序后")    for _, person := range persons {        fmt.Println(person.Name, ":", person.Age)    }}

其实,一般 Len() 和 Swap() 基本不做改变,只有涉及到元素比较的 Less() 方法会有所改变。

当我们对某一个结构体中多个字段进行排序时怎么办,难道每排序一个就写下这三个方法么,当然不是。我们可以利用嵌套结构体来解决这个问题。因为嵌套结构体可以继承父结构体的所有属性和方法
比如我想对上面 Person 的 Name 字段和 Age 对要排序,我们可以利用嵌套结构体来改进一下。

package mainimport (    "fmt"    "sort")type Person struct {    Name string    Age  int}type Persons []Person// Len()方法和Swap()方法不用变化// 获取此 slice 的长度func (p Persons) Len() int { return len(p) }// 交换数据func (p Persons) Swap(i, j int) { p[i], p[j] = p[j], p[i] }// 嵌套结构体  将继承 Person 的所有属性和方法// 所以相当于SortByName 也实现了 Len() 和 Swap() 方法type SortByName struct{ Persons }// 根据元素的姓名长度降序排序 (此处按照自己的业务逻辑写)func (p SortByName) Less(i, j int) bool {    return len(p.Persons[i].Name) > len(p.Persons[j].Name)}type SortByAge struct{ Persons }// 根据元素的年龄降序排序 (此处按照自己的业务逻辑写)func (p SortByAge) Less(i, j int) bool {    return p.Persons[i].Age > p.Persons[j].Age}func main() {    persons := Persons{        {            Name: "test123",            Age:  20,        },        {            Name: "test1",            Age:  22,        },        {            Name: "test12",            Age:  21,        },    }    fmt.Println("排序前")    for _, person := range persons {        fmt.Println(person.Name, ":", person.Age)    }    sort.Sort(SortByName{persons})    fmt.Println("排序后")    for _, person := range persons {        fmt.Println(person.Name, ":", person.Age)    }}

转载地址:http://vbmwx.baihongyu.com/

你可能感兴趣的文章
如何实现邀请好友帮抢票功能?
查看>>
深圳联通特邀湖北籍企业参观公司总部大楼举行
查看>>
告警系统主脚本、告警系统配置文件、告警系统监控项目
查看>>
Python 和 PyCharm 在 windows10 环境的安装和设置
查看>>
C语言入门基础之数组——数学和编程的完美结合(图)
查看>>
《远见》的读后感作文1000字范文
查看>>
重置密码、单用户模式、救援模式
查看>>
LAMP环境搭建1-mysql5.5
查看>>
第三课 Linux目录及文件管理、用户组管理及bash重定向
查看>>
shell 脚本攻略--小试牛刀
查看>>
spring boot view override
查看>>
bzoj 2282: [Sdoi2011]消防
查看>>
我的友情链接
查看>>
centos5.9使用RPM包搭建lamp平台
查看>>
关于C#面向对象2
查看>>
Javascript String类的属性及方法
查看>>
vim编辑器如何添加或删除多行注释
查看>>
[LeetCode] Merge Intervals
查看>>
iOS开发-按钮的基本使用
查看>>
在QT和SDL搭建的框架中使用OPENGL在SDL窗口上进行绘图
查看>>