Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > golang 对象list排序

golang 使用sort.slice包实现对象list排序

作者:峰啊疯了

这篇文章主要介绍了golang 使用sort.slice包实现对象list排序,对比sort跟slice两种排序的使用方式区别展开内容,需要的小伙伴可以参考一下

1.sort.Sort介绍

使用sort.Slice进行排序,因为slice把struct抽象化了,且slice封装过了,简单的基础类型可以使用sort,使用sort排序需要重写三个interface,不想学习sort排序的可以直接看第三步

这里将对比sort跟slice两种排序的使用方式

1.1分析内置sort包

#yyds干货盘点#golag 用sort.slice包实现对象list排序_字段

1.2分析sort.go

#yyds干货盘点#golag 用sort.slice包实现对象list排序_字段_02

要想实现自定义对象的排序功能,需要重写这三个interface

外部调用sort方法即可实现排序

#yyds干货盘点#golag 用sort.slice包实现对象list排序_字段_03

 
func Sort(data Interface) {
n := data.Len()
quickSort(data, 0, n, maxDepth(n))
}

通过sort源码可以看到用的快速排序,不懂快排的可以自行重温快排算法,这里就不讲解快排原理了

2.使用方法

2.1基础类型排序

基础类型的排序直接在sort.go去找就行了

package main

import (
"fmt"
"sort"
)

func main() {
intList := [] int {2, 4, 3, 5, 7, 6, 9, 8, 1, 0}
float8List := [] float64 {4.2, 5.9, 12.3, 10.0, 50.4, 99.9, 31.4, 27.81828, 3.14}
stringList := [] string {"a", "c", "b", "d", "f", "i", "z", "x", "w", "y"}

sort.Ints(intList)
sort.Float64s(float8List)
sort.Strings(stringList)

fmt.Printf("%v\n%v\n%v\n", intList, float8List, stringList)

}

#yyds干货盘点#golag 用sort.slice包实现对象list排序_字段_04

2.2对象排序(单一字段)

如果对一个struct对象的某一个字段进行排序就需要重写一组len,less,swap接口,如果有多组就需要重写多组接口。

下面直接看例子:

如果一个对象只有一个字段需要排序就用下面的例子比较简单

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Age int
}

// 按照 Person.Age 从大到小排序
type PersonSlice [] Person

func (a PersonSlice) Len() int { // 重写 Len() 方法
return len(a)
}
func (a PersonSlice) Swap(i, j int){ // 重写 Swap() 方法
a[i], a[j] = a[j], a[i]
}
func (a PersonSlice) Less(i, j int) bool { // 重写 Less() 方法, 从大到小排序
return a[j].Age < a[i].Age
}

func main() {
people := [] Person{
{"zhang san", 12},
{"li si", 30},
{"wang wu", 52},
{"zhao liu", 26},
}

fmt.Println(people)

sort.Sort(PersonSlice(people)) // 按照 Age 的逆序排序
fmt.Println(people)

sort.Sort(sort.Reverse(PersonSlice(people))) // 按照 Age 的升序排序
fmt.Println(people)

}

2.3对象排序(多字段)

多个了SortBy方法可以确定是升序还是降序,同时可以确定排序字段是哪个

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Age int
}

type PersonWrapper struct {
people [] Person
by func(p, q * Person) bool
}
//用来判断升序还是降序
type SortBy func(p, q *Person) bool

func (pw PersonWrapper) Len() int { // 重写 Len() 方法
return len(pw.people)
}
func (pw PersonWrapper) Swap(i, j int){ // 重写 Swap() 方法
pw.people[i], pw.people[j] = pw.people[j], pw.people[i]
}
func (pw PersonWrapper) Less(i, j int) bool { // 重写 Less() 方法
return pw.by(&pw.people[i], &pw.people[j])
}

// 封装成 SortPerson 方法
func SortPerson(people [] Person, by SortBy){
sort.Sort(PersonWrapper{people, by})
}

func main() {
people := [] Person{
{"zhang san", 12},
{"li si", 30},
{"wang wu", 52},
{"zhao liu", 26},
}

fmt.Println(people)

//推荐用封装的SortPerson方法
sort.Sort(PersonWrapper{people, func (p, q *Person) bool {
return q.Age < p.Age // Age 递减排序
}})

fmt.Println(people)

//推荐用这种
SortPerson(people, func (p, q *Person) bool {
return p.Name < q.Name // Name 递增排序
})

fmt.Println(people)
}

如果不喜欢用sortby方法,也可以重写多个less、len、swap方法,

下面的方法就调用起来比较简单,不用重写by跟sortby方法,但是上面的比较灵活(理解可能比下面的费劲)

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Weight int
}

type PersonSlice []Person

func (s PersonSlice) Len() int { return len(s) }
func (s PersonSlice) Swap(i, j int) { s[i], s[j] = s[j], s[i] }

type ByName struct{ PersonSlice } // 将 PersonSlice 包装起来到 ByName 中

func (s ByName) Less(i, j int) bool { return s.PersonSlice[i].Name < s.PersonSlice[j].Name } // 将 Less 绑定到 ByName 上


type ByWeight struct{ PersonSlice } // 将 PersonSlice 包装起来到 ByWeight 中
func (s ByWeight) Less(i, j int) bool { return s.PersonSlice[i].Weight < s.PersonSlice[j].Weight } // 将 Less 绑定到 ByWeight 上

func main() {
s := []Person{
{"apple", 12},
{"pear", 20},
{"banana", 50},
{"orange", 87},
{"hello", 34},
{"world", 43},
}

sort.Sort(ByWeight{s})
fmt.Println("People by weight:")
printPeople(s)

sort.Sort(ByName{s})
fmt.Println("\nPeople by name:")
printPeople(s)

}

func printPeople(s []Person) {
for _, o := range s {
fmt.Printf("%-8s (%v)\n", o.Name, o.Weight)
}
}

3.sort.Slice介绍

3.1使用方法

#yyds干货盘点#golag 用sort.slice包实现对象list排序_升序_05

很明显将对象抽象成interface,而上面对对象的排序,只能是person这种特定对象

调用的时候有点类似java的重写compare方法

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Weight int
}


func main() {
s := []Person{
{"apple", 12},
{"pear", 20},
{"banana", 50},
{"orange", 87},
{"hello", 34},
{"world", 43},
}

//可以向上面一样对排序字段封装,加一个sortby字段传递
sort.Slice(s,func(i,j int)bool{
return s[i].Weight < s[j].Weight
})
fmt.Println("People by weight:")
printPeople(s)

sort.Slice(s,func(i,j int)bool{
return s[i].Name < s[j].Name
})
fmt.Println("\nPeople by name:")
printPeople(s)

}

func printPeople(s []Person) {
for _, o := range s {
fmt.Printf("%-8s (%v)\n", o.Name, o.Weight)
}
}

3.2运行

#yyds干货盘点#golag 用sort.slice包实现对象list排序_封装_06

到此这篇关于golang 使用sort.slice包实现对象list排序的文章就介绍到这了,更多相关golang 对象list排序内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文