代码改变世界

go语言从例子开始之Example39.使用函数自定义排序

2019-11-01 09:04  _天枢  阅读(305)  评论(0编辑  收藏  举报

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

Example:

package main
import (
    "fmt"
    "sort"
)

//为了在 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])
}


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

Result:

$ go run example.go
[ab abc abcd]

 

坐标: 上一个例子   下一个例子