Go语言排序sort的使用

已知一个的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, 0, 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, a, b)
            return
        }
        maxDepth--
        mlo, mhi := doPivot(data, a, b)
        // Avoiding recursion on the larger subproblem guarantees
        // a stack depth of at most lg(b-a).
        if mlo-a < b-mhi {
            quickSort(data, a, mlo, maxDepth)
            a = mhi // i.e., quickSort(data, mhi, b)
        } else {
            quickSort(data, mhi, b, maxDepth)
            b = mlo // i.e., quickSort(data, a, 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, a, 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),虽然他们结构是一样的。

Linux公社的RSS地址:https://www.linuxidc.com/rssFeed.aspx

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/0b536063cc18b2caaf8d128ad3e823bf.html