有时候,我们希望通过除了自然顺序以外的其他方式对集合进行排序。例如,假设我们想按字符串的长度而不是字母顺序对字符串进行排序。下面是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编程环境请参考:http://www.yiibai.com/go/go_environment.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]