您好,欢迎访问一九零五行业门户网

关于golang之排序使用

下面由golang教程栏目给大家介绍golang之排序使用,希望对需要的朋友有所帮助!
golang标准库实现了许多常用的排序方法,比如对整数序列排序:sort.ints(),
那么如果对自定义的数据结构排序怎么做呢?
比如对一个用户列表,按他们的积分排序:
首先定义数据结构,为了能清楚说明问题,只给两个字段。
type user struct { name string score int}type users []user
golang中想要自定义排序,自己的结构要实现三个方法:
// 摘自: $goroot/src/sort/sort.gotype interface interface { // len is the number of elements in the collection. len() int // less reports whether the element with // index i should sort before the element with index j. less(i, j int) bool // swap swaps the elements with indexes i and j. swap(i, j int)}
这个设计太妙了有没有,想想我们学过的排序,都要序列长度,比大小,交换元素。
那对上述的users,也就是用户列表如何使用golang的排序呢?
先按它说的,实现这三个方法:
func (us users) len() int { return len(us)}func (us users) less(i, j int) bool { return us[i].score < us[j].score}func (us users) swap(i, j int) { us[i], us[j] = us[j], us[i]}
然后就能排序了:
func main() { var us users const n = 6 for i := 0; i < n; i++ { us = append(us, user{ name: "user" + strconv.itoa(i), score: rand.intn(n * n), }) } fmt.printf("%v\n", us) sort.sort(us) fmt.printf("%v\n", us)}
可能的输出为:
[{user0 5} {user1 15} {user2 11} {user3 11} {user4 13} {user5 6}]
[{user0 5} {user5 6} {user2 11} {user3 11} {user4 13} {user1 15}]
可以看到,分数从小到大排列了。
不过一般我们积分这种东西都是从大到小排序的,只需将
sort.sort(us)改成sort.sort(sort.reverse(us))就行。
确实很方便。
当然,如果出于特殊需要,系统提供的排序不能满足我们的需要,
还是可以自己实现排序的, 比如针对上述,自己来排序(从小到大):
func myqsort(us []user, lo, hi int) { if lo < hi { pivot := partition(us, lo, hi) myqsort(us, lo, pivot-1) myqsort(us, pivot+1, hi) }}func partition(us []user, lo, hi int) int { tmp := us[lo] for lo < hi { for lo < hi && us[hi].score >= tmp.score { hi-- } us[lo] = us[hi] for lo < hi && us[lo].score <= tmp.score { lo++ } us[hi] = us[lo] } us[lo] = tmp return hi}
一个简单的快速排序,调用时只需要myqsort(us)就可以 了。
总结:
自定义序列要实现less, swap, len三个方法 才行欢迎补充指正!
以上就是关于golang之排序使用的详细内容。
其它类似信息

推荐信息