深入理解golang的基本类型排序与slice排序

前言

其实golang的排序思路和c和c++有些差别。 c默认是对数组进行排序， c++是对一个序列进行排序， go则更宽泛一些，待排序的可以是任何对象， 虽然很多情况下是一个slice(分片， 类似于数组)，或是包含 slice 的一个对象。

排序(接口)的三个要素：

      1、待排序元素个数 n ；

      2、第 i 和第 j 个元素的比较函数 cmp ；

      3、第 i 和 第 j 个元素的交换 swap ；

乍一看条件 3 是多余的， c 和 c++ 都不提供 swap 。 c 的 qsort 的用法： qsort(data, n, sizeof(int), cmp_int); data 是起始地址， n 是元素个数， sizeof(int) 是每个元素的大小， cmp_int 是一个比较两个 int 的函数。

c++ 的 sort 的用法： sort(data, data+n, cmp_int); data 是第一个元素的位置， data+n 是最后一个元素的下一个位置， cmp_int 是比较函数。

基本类型排序(int、float64 和 string)

1、升序排序

对于 int 、 float64 和 string 数组或是分片的排序， go 分别提供了 sort.ints() 、 sort.float64s() 和 sort.strings() 函数， 默认都是从小到大排序。

package main

import (
 "fmt"
 "sort"
)

func main() {
 intlist := [] int {2, 4, 3, 5, 7, 6, 9, 8, 1, 0}
 float8list := [] float64 {4.2, 5.9, 12.3, 10.0, 50.4, 99.9, 31.4, 27.81828, 3.14}
 stringlist := [] string {"a", "c", "b", "d", "f", "i", "z", "x", "w", "y"}

 sort.ints(intlist)
 sort.float64s(float8list)
 sort.strings(stringlist)

 fmt.printf("%v\n%v\n%v\n", intlist, float8list, stringlist)

}

2、降序排序

int 、 float64 和 string 都有默认的升序排序函数， 现在问题是如果降序如何 ？ 有其他语言编程经验的人都知道，只需要交换 cmp 的比较法则就可以了， go 的实现是类似的，然而又有所不同。 go 中对某个 type 的对象 obj 排序， 可以使用 sort.sort(obj) 即可，就是需要对 type 类型绑定三个方法 ： len() 求长度、less(i,j) 比较第 i 和 第 j 个元素大小的函数、 swap(i,j) 交换第 i 和第 j 个元素的函数。sort 包下的三个类型 intslice 、 float64slice 、 stringslice 分别实现了这三个方法， 对应排序的是 [] int 、 [] float64 和 [] string 。如果期望逆序排序， 只需要将对应的 less 函数简单修改一下即可。

go 的 sort 包可以使用 sort.reverse(slice) 来调换 slice.interface.less ，也就是比较函数，所以， int 、 float64 和 string 的逆序排序函数可以这么写：

package main

import (
 "fmt"
 "sort"
)

func main() {
 intlist := [] int {2, 4, 3, 5, 7, 6, 9, 8, 1, 0}
 float8list := [] float64 {4.2, 5.9, 12.3, 10.0, 50.4, 99.9, 31.4, 27.81828, 3.14}
 stringlist := [] string {"a", "c", "b", "d", "f", "i", "z", "x", "w", "y"}

 sort.sort(sort.reverse(sort.intslice(intlist)))
 sort.sort(sort.reverse(sort.float64slice(float8list)))
 sort.sort(sort.reverse(sort.stringslice(stringlist)))

 fmt.printf("%v\n%v\n%v\n", intlist, float8list, stringlist)
}

3、深入理解排序

sort 包中有一个 sort.interface 接口，该接口有三个方法 len()  、 less(i,j) 和 swap(i,j) 。 通用排序函数 sort.sort 可以排序任何实现了 sort.inferface 接口的对象(变量)。对于 [] int 、[] float64 和 [] string 除了使用特殊指定的函数外，还可以使用改装过的类型 intsclice 、 float64slice 和 stringslice ， 然后直接调用它们对应的 sort() 方法；因为这三种类型也实现了 sort.interface 接口， 所以可以通过 sort.reverse 来转换这三种类型的 interface.less 方法来实现逆向排序， 这就是前面最后一个排序的使用。

下面使用了一个自定义(用户定义)的 reverse 结构体， 而不是 sort.reverse 函数， 来实现逆向排序。

package main

import (
 "fmt"
 "sort"
)

// 自定义的 reverse 类型
type reverse struct {
 sort.interface // 这样，reverse可以接纳任何实现了sort.interface的对象
}

// reverse 只是将其中的 inferface.less 的顺序对调了一下
func (r reverse) less(i, j int) bool {
 return r.interface.less(j, i)
}

func main() {
 ints := []int{5, 2, 6, 3, 1, 4}

 sort.ints(ints)          // 特殊排序函数，升序
 fmt.println("after sort by ints:\t", ints)

 doubles := []float64{2.3, 3.2, 6.7, 10.9, 5.4, 1.8}

 sort.float64s(doubles)
 fmt.println("after sort by float64s:\t", doubles) // [1.8 2.3 3.2 5.4 6.7 10.9]

 strings := []string{"hello", "good", "students", "morning", "people", "world"}
 sort.strings(strings)
 fmt.println("after sort by strings:\t", strings) // [good hello mornig people students world]

 ipos := sort.searchints(ints, -1) // int 搜索
 fmt.printf("pos of 5 is %d th\n", ipos)

 dpos := sort.searchfloat64s(doubles, 20.1) // float64 搜索
 fmt.printf("pos of 5.0 is %d th\n", dpos)

 fmt.printf("doubles is asc ? %v\n", sort.float64saresorted(doubles))

 doubles = []float64{3.5, 4.2, 8.9, 100.98, 20.14, 79.32}
 // sort.sort(sort.float64slice(doubles)) // float64 排序方法 2
 // fmt.println("after sort by sort:\t", doubles) // [3.5 4.2 8.9 20.14 79.32 100.98]
 (sort.float64slice(doubles)).sort()   // float64 排序方法 3
 fmt.println("after sort by sort:\t", doubles)  // [3.5 4.2 8.9 20.14 79.32 100.98]

 sort.sort(reverse{sort.float64slice(doubles)}) // float64 逆序排序
 fmt.println("after sort by reversed sort:\t", doubles)  // [100.98 79.32 20.14 8.9 4.2 3.5]
}

sort.ints / sort.float64s / sort.strings 分别来对整型/浮点型/字符串型slice进行排序。然后是有个测试是否有序的函数。还有分别对应的 search 函数，不过，发现搜索函数只能定位到如果存在的话的位置，不存在的话，位置是不对的。

关于一般的数组排序，程序中显示了，有 3 种方法！目前提供的三种类型 int，float64 和 string 呈现对称的，也就是你有的，对应的我也有。关于翻转排序或是逆向排序，就是用个翻转结构体，重写 less() 函数即可。

上面的 reverse 是个通用的结构体。

上面说了那么多， 只是对基本类型进行排序， 该到说说 struct 结构体类型的排序的时候了， 实际中这个用得到的会更多。

结构体类型的排序

结构体类型的排序是通过使用 sort.sort(slice) 实现的， 只要 slice 实现了 sort.interface 的三个方法就可以。 虽然这么说，但是排序的方法却有那么好几种。首先一种就是模拟排序 [] int 构造对应的 intslice 类型，然后对 intslice 类型实现 interface 的三个方法。

1、模拟 intslice 排序

package main

import (
 "fmt"
 "sort"
)

type person struct {
 name string
 age int
}

// 按照 person.age 从大到小排序
type personslice [] person

func (a personslice) len() int {   // 重写 len() 方法
 return len(a)
}
func (a personslice) swap(i, j int){  // 重写 swap() 方法
 a[i], a[j] = a[j], a[i]
}
func (a personslice) less(i, j int) bool { // 重写 less() 方法， 从大到小排序
 return a[j].age < a[i].age
}

func main() {
 people := [] person{
  {"zhang san", 12},
  {"li si", 30},
  {"wang wu", 52},
  {"zhao liu", 26},
 }

 fmt.println(people)

 sort.sort(personslice(people)) // 按照 age 的逆序排序
 fmt.println(people)

 sort.sort(sort.reverse(personslice(people))) // 按照 age 的升序排序
 fmt.println(people)

}

这完全是一种模拟的方式，所以如果懂了 intslice 自然就理解这里了，反过来，理解了这里那么 intslice 那里也就懂了。

这种方法的缺点是：根据 age 排序需要重新定义 personslice 方法，绑定 len 、 less 和 swap 方法， 如果需要根据 name 排序， 又需要重新写三个函数； 如果结构体有 4 个字段，有四种类型的排序，那么就要写 3 × 4 = 12 个方法， 即使有一些完全是多余的， o__o”… 仔细思量一下，根据不同的标准 age 或是 name， 真正不同的体现在 less 方法上，所以可以将 less 抽象出来， 每种排序的 less 让其变成动态的，比如下面一种方法。

2、封装成 wrapper

package main

import (
 "fmt"
 "sort"
)

type person struct {
 name string
 age int
}

type personwrapper struct {     //注意此处
 people [] person
 by func(p, q * person) bool
}

func (pw personwrapper) len() int {   // 重写 len() 方法
 return len(pw.people)
}
func (pw personwrapper) swap(i, j int){  // 重写 swap() 方法
 pw.people[i], pw.people[j] = pw.people[j], pw.people[i]
}
func (pw personwrapper) less(i, j int) bool { // 重写 less() 方法
 return pw.by(&pw.people[i], &pw.people[j])
}

func main() {
 people := [] person{
  {"zhang san", 12},
  {"li si", 30},
  {"wang wu", 52},
  {"zhao liu", 26},
 }

 fmt.println(people)

 sort.sort(personwrapper{people, func (p, q *person) bool {
  return q.age < p.age // age 递减排序
 }})

 fmt.println(people)
 sort.sort(personwrapper{people, func (p, q *person) bool {
  return p.name < q.name // name 递增排序
 }})

 fmt.println(people)

}

这种方法将 [] person 和比较的准则 cmp 封装在了一起，形成了 personwrapper 函数，然后在其上绑定 len 、 less 和 swap 方法。 实际上 sort.sort(pw) 排序的是 pw 中的 people， 这就是前面说的， go 的排序未必就是针对的一个数组或是 slice， 而可以是一个对象中的数组或是 slice 。

3、进一步封装

感觉方法 2 已经很不错了， 唯一一个缺点是，在 main 中使用的时候暴露了 sort.sort 的使用，还有就是 personwrapper 的构造。 为了让 main 中使用起来更为方便， me 们可以再简单的封装一下， 构造一个 sortperson 方法， 如下：

package main

import (
 "fmt"
 "sort"
)

type person struct {
 name string
 age int
}

type personwrapper struct {
 people [] person
 by func(p, q * person) bool
}

type sortby func(p, q *person) bool

func (pw personwrapper) len() int {   // 重写 len() 方法
 return len(pw.people)
}
func (pw personwrapper) swap(i, j int){  // 重写 swap() 方法
 pw.people[i], pw.people[j] = pw.people[j], pw.people[i]
}
func (pw personwrapper) less(i, j int) bool { // 重写 less() 方法
 return pw.by(&pw.people[i], &pw.people[j])
}

// 封装成 sortperson 方法
func sortperson(people [] person, by sortby){
 sort.sort(personwrapper{people, by})
}

func main() {
 people := [] person{
  {"zhang san", 12},
  {"li si", 30},
  {"wang wu", 52},
  {"zhao liu", 26},
 }

 fmt.println(people)

 sort.sort(personwrapper{people, func (p, q *person) bool {
  return q.age < p.age // age 递减排序
 }})

 fmt.println(people)

 sortperson(people, func (p, q *person) bool {
  return p.name < q.name // name 递增排序
 })

 fmt.println(people)

}

在方法 2 的基础上构造了 sortperson 函数，使用的时候传过去一个 [] person 和一个 cmp 函数。

4、另一种思路

package main

import (
 "fmt"
 "sort"
)

type person struct {
 name  string
 weight  int
}

type personslice []person

func (s personslice) len() int { return len(s) }
func (s personslice) swap(i, j int)  { s[i], s[j] = s[j], s[i] }

type byname struct{ personslice } // 将 personslice 包装起来到 byname 中

func (s byname) less(i, j int) bool  { return s.personslice[i].name < s.personslice[j].name } // 将 less 绑定到 byname 上


type byweight struct{ personslice } // 将 personslice 包装起来到 byweight 中
func (s byweight) less(i, j int) bool { return s.personslice[i].weight < s.personslice[j].weight } // 将 less 绑定到 byweight 上

func main() {
 s := []person{
  {"apple", 12},
  {"pear", 20},
  {"banana", 50},
  {"orange", 87},
  {"hello", 34},
  {"world", 43},
 }

 sort.sort(byweight{s})
 fmt.println("people by weight:")
 printpeople(s)

 sort.sort(byname{s})
 fmt.println("\npeople by name:")
 printpeople(s)

}

func printpeople(s []person) {
 for _, o := range s {
  fmt.printf("%-8s (%v)\n", o.name, o.weight)
 }
}

对结构体的排序， 暂时就到这里。 第一种排序对只根据一个字段的比较合适， 另外三个是针对可能根据多个字段排序的。方法 4 我认为每次都要多构造一个 byxxx ， 颇为不便， 这样多麻烦，不如方法 2 和方法 3 来的方便，直接传进去一个 cmp。 方法2、 3 没有太大的差别， 3 只是简单封装了一下而已， 对于使用者来说， 可能会更方便一些，而且也会更少的出错。

总结

以上就是关于golang基本类型排序与slice排序的全部内容，希望这篇文章的内容对啊大家学习或者使用golang能有所帮助，如果有疑问大家也可以留言交流，小编会尽快给大家回复的。