原文鏈接： Go 語言中排序的 3 種方法

在寫代碼過程中，排序是經(jīng)常會遇到的需求，本文會介紹三種常用的方法。

廢話不多說，下面正文開始。

使用標(biāo)準(zhǔn)庫

根據(jù)場景直接使用標(biāo)準(zhǔn)庫中的方法，比如：

• sort.Ints
• sort.Float64s
• sort.Strings

舉個例子：

s := []int{4, 2, 3, 1}
sort.Ints(s)
fmt.Println(s) // [1 2 3 4]

自定義比較器

使用 sort.Slice 方法排序時，可以自定義比較函數(shù) less(i, j int) bool，這樣就可以根據(jù)需要按不同的字段進行排序。

如果想要穩(wěn)定排序的話，就使用 sort.SliceStable 方法。

舉個例子：

family := []struct {
    Name string
    Age  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}]

自定義數(shù)據(jù)結(jié)構(gòu)

使用 sort.Sort 或者 sort.Stable 方法，它們可以對任意實現(xiàn)了 sort.Interface 的數(shù)據(jù)結(jié)構(gòu)排序。

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)
}

意思就是說，只要某一個數(shù)據(jù)結(jié)構(gòu)實現(xiàn)了 Len() int，Less(i, j int) bool 和 Swap(i, j int) 這三個方法，那么就可以使用 sort.Sort 來排序。

舉個例子：

type Person struct {
    Name string
    Age  int
}

// ByAge implements sort.Interface based on the Age field.
type ByAge []Person

func (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

以上就是本文的全部內(nèi)容，如果覺得還不錯的話歡迎點贊，轉(zhuǎn)發(fā)和關(guān)注，感謝支持。

參考文章：

• https://yourbasic.org/golang/how-to-sort-in-go/#performance-and-implementation

推薦閱讀：

• Go 語言 map 是并發(fā)安全的嗎？
• Go 語言切片是如何擴容的？
• Go 語言數(shù)組和切片的區(qū)別
• Go 語言 new 和 make 關(guān)鍵字的區(qū)別
• 為什么 Go 不支持 []T 轉(zhuǎn)換為 []interface
• 為什么 Go 語言 struct 要使用 tags