以下由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)
就可以了。
總結:
歡迎補充修正!
#以上是關於golang之排序使用的詳細內容。更多資訊請關注PHP中文網其他相關文章!