golag 使用sort.slice包實現物件list排序

2022-03-25 16:01:06

1.sort.Sort介紹

使用sort.Slice進行排序，因為slice把struct抽象化了，且slice封裝過了，簡單的基礎型別可以使用sort,使用sort排序需要重寫三個interface,不想學習sort排序的可以直接看第三步

這裡將對比sort跟slice兩種排序的使用方式

1.1分析內建sort包

1.2分析sort.go

要想實現自定義物件的排序功能，需要重寫這三個interface

外部呼叫sort方法即可實現排序

func Sort(data Interface) {
n := data.Len()
quickSort(data, 0, n, maxDepth(n))
}

通過sort原始碼可以看到用的快速排序，不懂快排的可以自行重溫快排演演算法，這裡就不講解快排原理了

2.使用方法

2.1基礎型別排序

基礎型別的排序直接在sort.go去找就行了

package main

import (
"fmt"
"sort"
)

func main() {
intList := [] int {2, 4, 3, 5, 7, 6, 9, 8, 1, 0}
float8List := [] float64 {4.2, 5.9, 12.3, 10.0, 50.4, 99.9, 31.4, 27.81828, 3.14}
stringList := [] string {"a", "c", "b", "d", "f", "i", "z", "x", "w", "y"}

sort.Ints(intList)
sort.Float64s(float8List)
sort.Strings(stringList)

fmt.Printf("%vn%vn%vn", intList, float8List, stringList)

}

2.2物件排序（單一欄位）

如果對一個struct物件的某一個欄位進行排序就需要重寫一組len,less,swap介面，如果有多組就需要重寫多組介面。

下面直接看例子：

如果一個物件只有一個欄位需要排序就用下面的例子比較簡單

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Age int
}

// 按照 Person.Age 從大到小排序
type PersonSlice [] Person

func (a PersonSlice) Len() int { // 重寫 Len() 方法
return len(a)
}
func (a PersonSlice) Swap(i, j int){ // 重寫 Swap() 方法
a[i], a[j] = a[j], a[i]
}
func (a PersonSlice) Less(i, j int) bool { // 重寫 Less() 方法， 從大到小排序
return a[j].Age < a[i].Age
}

func main() {
people := [] Person{
{"zhang san", 12},
{"li si", 30},
{"wang wu", 52},
{"zhao liu", 26},
}

fmt.Println(people)

sort.Sort(PersonSlice(people)) // 按照 Age 的逆序排序
fmt.Println(people)

sort.Sort(sort.Reverse(PersonSlice(people))) // 按照 Age 的升序排序
fmt.Println(people)

}

2.3物件排序（多欄位）

多個了SortBy方法可以確定是升序還是降序，同時可以確定排序欄位是哪個

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Age int
}

type PersonWrapper struct {
people [] Person
by func(p, q * Person) bool
}
//用來判斷升序還是降序
type SortBy func(p, q *Person) bool

func (pw PersonWrapper) Len() int { // 重寫 Len() 方法
return len(pw.people)
}
func (pw PersonWrapper) Swap(i, j int){ // 重寫 Swap() 方法
pw.people[i], pw.people[j] = pw.people[j], pw.people[i]
}
func (pw PersonWrapper) Less(i, j int) bool { // 重寫 Less() 方法
return pw.by(&pw.people[i], &pw.people[j])
}

// 封裝成 SortPerson 方法
func SortPerson(people [] Person, by SortBy){
sort.Sort(PersonWrapper{people, by})
}

func main() {
people := [] Person{
{"zhang san", 12},
{"li si", 30},
{"wang wu", 52},
{"zhao liu", 26},
}

fmt.Println(people)

//推薦用封裝的SortPerson方法
sort.Sort(PersonWrapper{people, func (p, q *Person) bool {
return q.Age < p.Age // Age 遞減排序
}})

fmt.Println(people)

//推薦用這種
SortPerson(people, func (p, q *Person) bool {
return p.Name < q.Name // Name 遞增排序
})

fmt.Println(people)
}

如果不喜歡用sortby方法，也可以重寫多個less、len、swap方法，

下面的方法就呼叫起來比較簡單，不用重寫by跟sortby方法，但是上面的比較靈活（理解可能比下面的費勁）

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Weight int
}

type PersonSlice []Person

func (s PersonSlice) Len() int { return len(s) }
func (s PersonSlice) Swap(i, j int) { s[i], s[j] = s[j], s[i] }

type ByName struct{ PersonSlice } // 將 PersonSlice 包裝起來到 ByName 中

func (s ByName) Less(i, j int) bool { return s.PersonSlice[i].Name < s.PersonSlice[j].Name } // 將 Less 繫結到 ByName 上


type ByWeight struct{ PersonSlice } // 將 PersonSlice 包裝起來到 ByWeight 中
func (s ByWeight) Less(i, j int) bool { return s.PersonSlice[i].Weight < s.PersonSlice[j].Weight } // 將 Less 繫結到 ByWeight 上

func main() {
s := []Person{
{"apple", 12},
{"pear", 20},
{"banana", 50},
{"orange", 87},
{"hello", 34},
{"world", 43},
}

sort.Sort(ByWeight{s})
fmt.Println("People by weight:")
printPeople(s)

sort.Sort(ByName{s})
fmt.Println("nPeople by name:")
printPeople(s)

}

func printPeople(s []Person) {
for _, o := range s {
fmt.Printf("%-8s (%v)n", o.Name, o.Weight)
}
}

3.sort.Slice介紹

3.1使用方法

很明顯將物件抽象成interface,而上面對物件的排序，只能是person這種特定物件

呼叫的時候有點類似java的重寫compare方法

package main

import (
"fmt"
"sort"
)

type Person struct {
Name string
Weight int
}


func main() {
s := []Person{
{"apple", 12},
{"pear", 20},
{"banana", 50},
{"orange", 87},
{"hello", 34},
{"world", 43},
}

//可以向上面一樣對排序欄位封裝，加一個sortby欄位傳遞
sort.Slice(s,func(i,j int)bool{
return s[i].Weight < s[j].Weight
})
fmt.Println("People by weight:")
printPeople(s)

sort.Slice(s,func(i,j int)bool{
return s[i].Name < s[j].Name
})
fmt.Println("nPeople by name:")
printPeople(s)

}

func printPeople(s []Person) {
for _, o := range s {
fmt.Printf("%-8s (%v)n", o.Name, o.Weight)
}
}