zoukankan      html  css  js  c++  java
  • 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),虽然他们结构是一样的。

    最近诸事不顺,心烦的一匹,上周也拖更了,总之希望大家事事顺心,心爱的人身体健康。

    希望对大家有所帮助~

  • 相关阅读:
    codevs 2010 求后序遍历
    code vs 1013 求先序排列
    codevs 3143 二叉树的序遍历
    codevs 3083 二叉树
    找树的根和孩子
    1501 二叉树最大宽度和高度
    1758:二叉树
    sql 如何把查询得到的结果如何放入一个新表中
    2011的n次方
    计算2的N次方
  • 原文地址:https://www.cnblogs.com/baiyb/p/9163818.html
Copyright © 2011-2022 走看看