go语言中sort包的实现方法与应用详解

前言

go语言的 sort 包实现了内置和用户定义类型的排序，sort包中实现了３种基本的排序算法：插入排序．快排和堆排序．和其他语言中一样，这三种方式都是不公开的，他们只在sort包内部使用．所以用户在使用sort包进行排序时无需考虑使用那种排序方式，sort.interface定义的三个方法：获取数据集合长度的len()方法、比较两个元素大小的less()方法和交换两个元素位置的swap()方法，就可以顺利对数据集合进行排序。sort包会根据实际数据自动选择高效的排序算法。

之前跟大家分享了go语言使用sort包对任意类型元素的集合进行排序的方法，感兴趣的朋友们可以参考这篇文章：

下面来看看sort包的简单示例：

type interface interface {
 // 返回要排序的数据长度
 len() int
 //比较下标为i和j对应的数据大小，可自己控制升序和降序  
less(i, j int) bool
 // 交换下标为i，j对应的数据
 swap(i, j int)
}

任何实现了 sort.interface 的类型（一般为集合），均可使用该包中的方法进行排序。这些方法要求集合内列出元素的索引为整数。

这里我直接用源码来讲解实现：

1、源码中的例子：

type person struct {
 name string
 age int
}

type byage []person
//实现了sort接口中的三个方法，则可以使用排序方法了
func (a byage) len() int   { return len(a) }
func (a byage) swap(i, j int)  { a[i], a[j] = a[j], a[i] }
func (a byage) less(i, j int) bool { return a[i].age < a[j].age }

func example() {
 people := []person{
  {"bob", 31},
  {"john", 42},
  {"michael", 17},
  {"jenny", 26},
 }

 fmt.println(people)
 sort.sort(byage(people)) //此处调用了sort包中的sort()方法,我们看一下这个方法
 fmt.println(people)

 // output:
 // [bob: 31 john: 42 michael: 17 jenny: 26]
 // [michael: 17 jenny: 26 bob: 31 john: 42]
}

2、sort(data interface)方法

//sort包只提供了这一个公开的公使用的排序方法，
func sort(data interface) {
 // switch to heapsort if depth of 2*ceil(lg(n+1)) is reached.
 //如果元素深度达到2*ceil(lg(n+1))则选用堆排序
 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) //堆排序方法，a=0,b=n
   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)
 }
}

//堆排序
func heapsort(data interface, a, b int) {
 first := a
 lo := 0
 hi := b - a

 // build heap with greatest element at top.
 //构建堆结构，最大的元素的顶部，就是构建大根堆
 for i := (hi - 1) / 2; i >= 0; i-- {
  siftdown(data, i, hi, first)
 }

 // pop elements, largest first, into end of data.
 //把first插入到data的end结尾
 for i := hi - 1; i >= 0; i-- {
  data.swap(first, first+i) //数据交换
  siftdown(data, lo, i, first) //堆重新筛选
 }
}

// siftdown implements the heap property on data[lo, hi).
// first is an offset into the array where the root of the heap lies.
func siftdown(data interface, lo, hi, first int) {
 //hi为数组的长度
 //这里有一种做法是把跟元素给取到存下来，但是为了方法更抽象，省掉了这部，取而代之的是在swap的时候进行相互交换
 root := lo //根元素的下标
 for {
  child := 2*root + 1 //左叶子结点下标
  //控制for循环介绍，这种写法更简洁，可以查看我写的堆排序的文章
  if child >= hi { 
   break
  }
  //防止数组下标越界，判断左孩子和右孩子那个大
  if child+1 < hi && data.less(first+child, first+child+1) { 
   child++
  }
  //判断最大的孩子和根元素之间的关系
  if !data.less(first+root, first+child) {
   return
  }
  //如果上面都 满足，则进行数据交换
  data.swap(first+root, first+child)
  root = child
 }
}

这个包中还有很多方法，这个包实现了很多方法，比如排序反转，二分搜索。排序通过 quicksort（）这个方法来控制该调用快排还是堆排。

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对的支持。