使用sort自定义排序规则

4年以前  |  阅读数:169 次  |  编程语言:Golang 

有时候我们想使用和集合的自然排序不同的方法对集合进行排序。例如,我们想按照字母的长度而不是首字母顺序对字符串排序。

这里是一个 Go 自定义排序的例子。

package main

import "sort"
import "fmt"

// 为了在 Go 中使用自定义函数进行排序,我们需要一个对应的
// 类型。这里我们创建一个为内置 `[]string` 类型的别名的
// `ByLength` 类型,
type ByLength []string

// 我们在类型中实现了 `sort.Interface` 的 `Len`,`Less`
// 和 `Swap` 方法,这样我们就可以使用 `sort` 包的通用
// `Sort` 方法了,`Len` 和 `Swap` 通常在各个类型中都差
// 不多,`Less` 将控制实际的自定义排序逻辑。在我们的例
// 子中,我们想按字符串长度增加的顺序来排序,所以这里
// 使用了 `len(s[i])` 和 `len(s[j])`。
func (s ByLength) Len() int {
    return len(s)
}
func (s ByLength) Swap(i, j int) {
    s[i], s[j] = s[j], s[i]
}
func (s ByLength) Less(i, j int) bool {
    return len(s[i]) < len(s[j])
}

// 一切都准备好了,我们现在可以通过将原始的 `fruits` 切片转型成 `ByLength` 来实现我们的自定排序了。然后对这个转型的切片使用 `sort.Sort` 方法。
func main() {
    fruits := []string{"peach", "banana", "kiwi"}
    sort.Sort(ByLength(fruits))
    fmt.Println(fruits)
}

运行这个程序,和预期的一样,显示了一个按照字符串长度排序的列表。

执行程序:

$ go run sorting-by-functions.go 
[kiwi peach banana]

类似的,参照这个创建一个自定义类型的方法,实现这个类型的这三个接口方法,然后在一个这个自定义类型的集合上调用 sort.Sort 方法,我们就可以使用任意的函数来排序 Go 切片了。

 相关文章:
Golang语法基础:可变参数
Golang使用定时器
Golang语法基础:在结构体类型中定义方法
使用errors传递错误信息
从字符串中解析数字
Golang使用`math/rand`包生成伪随机数
Golang语法基础:使用map
使用printf格式化字符串
使用匿名函数
定义与使用常量
Golang使用正则表达式
使用`strings`包处理字符串
使用`sort`包排序
Golang处理时间的格式化及其与字符串之间的转换
定义与使用函数
Golang语法基础:利用自动推导定义变量
Golang语法基础,使用不同类型的数据
Golang语法基础,使用if
Golang语法基础:使用协程
Golang语法基础:内建多返回值支持