欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

golang对自定义类型进行排序的解决方法

程序员文章站 2022-04-09 20:54:29
前言 go 语言支持我们自定义类型,我们大家在实际项目中,常常需要根据一个结构体类型的某个字段进行排序。之前遇到这个问题不知道如何解决,后来在网上搜索了相关问题,找到了一...

前言

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

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

sort包浅谈

golang中也实现了排序算法的包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 main
import (
 "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 main
import (
 "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)
 }
}

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对的支持。