原文鏈接: Go 語言中排序的 3 種方法
在寫代碼過程中,排序是經常會遇到的需求,本文會介紹三種常用的方法。
廢話不多說,下面正文開始。
使用標準庫
根據場景直接使用標準庫中的方法,比如:
sort.Ints
sort.Float64s
sort.Strings
舉個例子:
s := []int{4, 2, 3, 1}
sort.Ints(s)
fmt.Println(s) // [1 2 3 4]
自定義比較器
使用 sort.Slice
方法排序時,可以自定義比較函數 less(i, j int) bool
,這樣就可以根據需要按不同的字段進行排序。
如果想要穩定排序的話,就使用 sort.SliceStable
方法。
舉個例子:
family := []struct {Name stringAge int
}{{"Alice", 23},{"David", 2},{"Eve", 2},{"Bob", 25},
}// Sort by age, keeping original order or equal elements.
sort.SliceStable(family, func(i, j int) bool {return family[i].Age < family[j].Age
})
fmt.Println(family) // [{David 2} {Eve 2} {Alice 23} {Bob 25}]
自定義數據結構
使用 sort.Sort
或者 sort.Stable
方法,它們可以對任意實現了 sort.Interface
的數據結構排序。
type 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)
}
意思就是說,只要某一個數據結構實現了 Len() int
,Less(i, j int) bool
和 Swap(i, j int)
這三個方法,那么就可以使用 sort.Sort
來排序。
舉個例子:
type Person struct {Name stringAge int
}// ByAge implements sort.Interface based on the Age field.
type ByAge []Personfunc (a ByAge) Len() int { return len(a) }
func (a ByAge) Less(i, j int) bool { return a[i].Age < a[j].Age }
func (a ByAge) Swap(i, j int) { a[i], a[j] = a[j], a[i] }func main() {family := []Person{{"Alice", 23},{"Eve", 2},{"Bob", 25},}sort.Sort(ByAge(family))fmt.Println(family) // [{Eve 2} {Alice 23} {Bob 25}]
}
字典排序
我們都知道,字典是無序的,具體原因可以看之前寫的這篇文章 Go 語言 map 如何順序讀取?
如果想要字典按 key 或者 value 排序的話,可以這樣做。
m := map[string]int{"Alice": 2, "Cecil": 1, "Bob": 3}keys := make([]string, 0, len(m))
for k := range m {keys = append(keys, k)
}
sort.Strings(keys)for _, k := range keys {fmt.Println(k, m[k])
}
// Output:
// Alice 2
// Bob 3
// Cecil 1
以上就是本文的全部內容,如果覺得還不錯的話歡迎點贊,轉發和關注,感謝支持。
參考文章:
- https://yourbasic.org/golang/how-to-sort-in-go/#performance-and-implementation
推薦閱讀:
- Go 語言 map 是并發安全的嗎?
- Go 語言切片是如何擴容的?
- Go 語言數組和切片的區別
- Go 語言 new 和 make 關鍵字的區別
- 為什么 Go 不支持 []T 轉換為 []interface
- 為什么 Go 語言 struct 要使用 tags