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),虽然他们结构是一样的。

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

希望对大家有所帮助~

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/tech/pnotes/12477.html

(0)
上一篇 2021年7月19日 14:41
下一篇 2021年7月19日 14:42

相关推荐

发表回复

登录后才能评论