Go按自定义函数排序实例

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

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

需要实现sort.Interface - Len，Less和Swap - 在这个类型上，所以可以使用 sort 包中的一般Sort函数。Len和Swap通常在类型之间是相似的，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编程代码实例