Go语言之城 - 程序示例: 使用函数自定义排序

有时候，我们可能想根据自然顺序以外的方式来对集合进行排序。 例如，假设我们要按字符串的长度而不是按字母顺序对它们进行排序。 这儿有一个在 Go 中自定义排序的示例。

 
package main

import (
    "fmt"
    "sort"
)

为了在 Go 中使用自定义函数进行排序，我们需要一个对应的类型。 我们在这里创建了一个 byLength 类型，它只是内建类型 []string 的别名。

 
type byLength []string

我们为该类型实现了 sort.Interface 接口的 Len、Less 和 Swap 方法， 这样我们就可以使用 sort 包的通用 Sort 方法了， Len 和 Swap 在各个类型中的实现都差不多， Less 将控制实际的自定义排序逻辑。 在这个的例子中，我们想按字符串长度递增的顺序来排序， 所以这里使用了 len(s[i]) 和 len(s[j]) 来实现 Less。

 
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]

类似的，参照这个例子，创建一个自定义类型， 为它实现 Interface 接口的三个方法， 然后对这个自定义类型的切片调用 sort.Sort 方法， 我们就可以通过任意函数对 Go 切片进行排序了。

下一个例子: Panic

Go语言之城 | 闽ICP备20003902号-3