Go按自定义函数排序实例

有时候,我们希望通过除了自然顺序以外的其他方式对集合进行排序。例如,假设我们想按字符串的长度而不是字母顺序对字符串进行排序。下面是Go语言中自定义排序的示例。

为了使用Go语言中的自定义函数进行排序,我们需要一个相应的类型。这里创建了一个ByLength类型,它只是内置 []string类型的别名。

需要实现sort.Interface - LenLessSwap - 在这个类型上,所以可以使用 sort 包中的一般Sort函数。LenSwap通常在类型之间是相似的,Less保存实际的自定义排序逻辑。在这个例子中,要按照字符串长度的增加顺序排序,因此在这里使用len(s [i])len(s [j])

所有这些都到位后,现在可以通过将原始 fruits 切片转换为ByLength来实现自定义排序,然后对该类型切片使用sort.Sort()方法。

运行程序根据需要显示按字符串长度排序的列表。

通过遵循创建自定义类型的模式,在该类型上实现三个Interface方法,然后在自定义类型的集合上调用sort.Sort,可以通过任意函数对Go切片进行排序。

所有的示例代码,都放在 F:\worksp\golang 目录下。安装Go编程环境请参考:/tutorial/detail-5562.html

sorting-by-functions.go的完整代码如下所示 -

package main

import "sort"
import "fmt"

// In order to sort by a custom function in Go, we need a
// corresponding type. Here we've created a `ByLength`
// type that is just an alias for the builtin `[]string`
// type.
type ByLength []string

// We implement `sort.Interface` - `Len`, `Less`, and
// `Swap` - on our type so we can use the `sort` package's
// generic `Sort` function. `Len` and `Swap`
// will usually be similar across types and `Less` will
// hold the actual custom sorting logic. In our case we
// want to sort in order of increasing string length, so
// we use `len(s[i])` and `len(s[j])` here.
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])
}

// With all of this in place, we can now implement our
// custom sort by casting the original `fruits` slice to
// `ByLength`, and then use `sort.Sort` on that typed
// slice.
func main() {
    fruits := []string{"peach", "banana", "kiwi"}
    sort.Sort(ByLength(fruits))
    fmt.Println(fruits)
}

执行上面代码,将得到以下输出结果 -

F:\worksp\golang>go run sorting-by-functions.go
[kiwi peach banana]

上一篇:Go排序实例

下一篇:Go panic错误处理实例

关注微信小程序
程序员编程王-随时随地学编程

扫描二维码
程序员编程王

扫一扫关注最新编程教程