go 排序sort的使用
发布时间：2020-12-16 09:35:20 所属栏目：大数据 来源：网络整理
导读：已知一个的struct组成的数组，现在要按照数组中的一个字段排序。python有sort方法，那golang要怎么实现呢？其实golang也有sort方法，并且使用简单，功能强大。 我们先看一下sort.Sort()的源码 func Sort(data Interface) {// Switch to heapsort if depth of
|
已知一个的struct组成的数组，现在要按照数组中的一个字段排序。python有sort方法，那golang要怎么实现呢？其实golang也有sort方法，并且使用简单，功能强大。 我们先看一下sort.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,n,maxDepth)
}
func quickSort(data Interface,a,b,maxDepth int) {
for b-a > 12 { // Use ShellSort for slices <= 12 elements
if maxDepth == 0 {
heapSort(data,b)
return
}
maxDepth--
mlo,mhi := doPivot(data,b)
// Avoiding recursion on the larger subproblem guarantees
// a stack depth of at most lg(b-a).
if mlo-a < b-mhi {
quickSort(data,mlo,maxDepth)
a = mhi // i.e.,quickSort(data,mhi,b)
} else {
quickSort(data,maxDepth)
b = mlo // i.e.,mlo)
}
}
if b-a > 1 {
// Do ShellSort pass with gap 6
// It could be written in this simplified form cause b-a <= 12
for i := a + 6; i < b; i++ {
if data.Less(i,i-6) {
data.Swap(i,i-6)
}
}
insertionSort(data,b)
}
}
通过quickSort方法我们看出，根据数据不同的状况，quickSort会选择堆排，快排，插入排中更高效的排序方法，这个我们暂时先不深究，先看如何使用sort.Sort()。 从源码中我们发现数据data，需要自带Len，Less，Swap三种方法，所以使用sort.Sort()前，我们需要自己实现Len，Less，Swap来确定按照什么规则排序。 ? 来实践一下，我们先创建一个结构体数组，结构体包含字段Count，我们按照Count对其排序。 package main
import (
"sort"
)
type subInfo struct {
Count uint `json:"count"`
}
type SubList []*subInfo
func (p SubList) Swap(i,j int) { p[i],p[j] = p[j],p[i] }
func (p SubList) Len() int { return len(p) }
func (p SubList) Less(i,j int) bool { return p[i].Count > p[j].Count }
func main() {
st_list := SubList{}
for i := 0; i < 10; i ++{
_sub := &subInfo{
Count: uint(i),}
st_list = append(st_list,_sub)
}
sort.Sort(st_list)
return
}
? 需要注意的是，st_list只能初始化成SubList{}，而不能初始化为make([]*subInfo,0)，虽然他们结构是一样的。 最近诸事不顺，心烦的一匹，上周也拖更了，总之希望大家事事顺心，心爱的人身体健康。 希望对大家有所帮助～ （编辑：李大同） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |