Golang 語言怎么高效排序數據?
01介紹
在 Golang 語言項目開發中,經常會遇到數據排序問題。Golang 語言標準庫 sort 包,為我們提供了數據排序的功能,我們可以直接使用 sort.Sort() 函數進行數據排序,sort.Sort() 函數底層實現是以快排為主,并根據目標數據的具體情況選擇不同的排序算法。本文我們介紹 sort 包排序數據的使用方法。
02切片排序
在 Golang 語言標準庫 sort 包中,sort.Sort() 函數用于數據排序,該函數需要一個 interface 類型的入參 sort.Interface,它包含三個方法,分別是 Len(),Less() 和 Swap()。也就是說,如果我們需要使用 sort 包的 Sort 函數進行數據排序,首先入參的數據需要實現這三個方法,或者理解為任意元素類型的切片實現了這三個方法,都可以使用 sort.Sort() 函數排序數據。
sort 包代碼:
- type Interface interface {
- Len() int // 集合中元素的數量
- Less(i, j int) bool // 描述元素的順序
- Swap(i, j int) // 交換索引為 i 和 j 的元素
- }
- func Sort(data Interface)
需要注意的是 sort.Sort() 函數不能保證數據排序是穩定的,如果需要保證數據排序穩定,可以使用 sort.Stable() 函數,“穩定”的含義是原始數據中 a 和 b 的值相等,排序前 a 排在 b 的前面,排序后 a 仍排在 b 的前面。
為了方便讀者朋友們理解,我們使用 int 類型的切片作為示例,介紹 sort.Sort() 函數的使用方法,我們定義一個類型 type IntSlice []int,并且給類型 IntSlice 實現 sort.Interface 接口類型定義的三個方法,然后使用 sort.Sort() 函數排序數據。
示例代碼:
- package main
- import (
- "fmt"
- "sort"
- )
- type IntSlice []int
- func (s IntSlice) Len() int {
- return len(s)
- }
- func (s IntSlice) Less(i, j int) bool {
- return s[i] > s[j]
- }
- func (s IntSlice) Swap(i, j int) {
- s[i], s[j] = s[j], s[i]
- }
- func main () {
- intSlice := IntSlice([]int{1, 3, 5, 7, 9})
- fmt.Println(intSlice) // 排序前
- sort.Sort(intSlice)
- fmt.Println(intSlice) // 排序后
- }
輸出結構:
- [9 7 5 3 1]
- [1 3 5 7 9]
讀到這里,我相信聰明的讀者朋友們已經了解了 sort.Sort() 的使用方式,同時也會產生一個疑問,難道每次使用 sort.Sort() 排序數據,都需要這么麻煩嗎?我還不如自己寫個遍歷排序數據。
是的,當然不用這么麻煩,sort 包已經幫我們封裝好了常用函數,我們直接使用就可以了。所以,上面的示例代碼可以使用 sort.Ints() 函數排序數據。
示例代碼:
- func main () {
- intSlice := IntSlice([]int{9, 7, 5, 3, 1})
- fmt.Println(intSlice) // 排序前
- sort.Ints(intSlice)
- fmt.Println(intSlice) // 使用 sort.Ints() 排序數據
- }
除了 sort.Ints(),還有 sort.Float64s(),sort.Strings()等。
03自定義集合排序
在 Golang 語言項目開發中,我們經常會使用結構體,如果我們需要排序結構體類型的切片,應該怎么操作呢?
我們可以按照 Part 01 介紹的方式,實現那三個方法,然后調用 sort.Sort() 函數,當然,sort 包也為我們封裝了排序結構體類型切片的函數 sort.Slice(),但是,參數除了需要排序的數據之外,還需要提供一個 Less() 函數類型的參數。
示例代碼:
- people := []struct {
- Name string
- Age int
- }{
- {"Gopher", 7},
- {"Alice", 55},
- {"Vera", 24},
- {"Bob", 75},
- }
- sort.Slice(people, func(i, j int) bool { return people[i].Name < people[j].Name })
- fmt.Println("By name:", people)
- sort.Slice(people, func(i, j int) bool { return people[i].Age < people[j].Age })
- fmt.Println("By age:", people)
輸出結果:
- By name: [{Alice 55} {Bob 75} {Gopher 7} {Vera 24}]
- By age: [{Gopher 7} {Vera 24} {Alice 55} {Bob 75}]
04總結
本文我們介紹了怎么使用 Golang 語言標準庫 sort 包排序數據,需要注意的是,除了本文使用的類型之外,其它任意類型只要實現 sort.Interface 的三個方法,都可以調用 sort.Sort() 函數排序數據。
另外,除了排序數據之外,sort 包也為我們提供了查找功能的函數 sort.Search(),感興趣的讀者朋友們可以閱讀 Golang 語言官方標準庫文檔了解更多。
本文轉載自微信公眾號「Golang語言開發棧」,可以通過以下二維碼關注。轉載本文請聯系Golang語言開發棧公眾號。



























