golang對(duì)自定義類型進(jìn)行排序的解決方法
前言
Go 語言支持我們自定義類型,我們大家在實(shí)際項(xiàng)目中,常常需要根據(jù)一個(gè)結(jié)構(gòu)體類型的某個(gè)字段進(jìn)行排序。之前遇到這個(gè)問題不知道如何解決,后來在網(wǎng)上搜索了相關(guān)問題,找到了一些好的解決方案,此處參考下,做個(gè)總結(jié)吧。
由于 golang 的 sort 包本身就提供了相應(yīng)的功能, 我們就沒必要重復(fù)的造個(gè)輪子了,來看看如何利用 sort 包來實(shí)現(xiàn)吧。
sort包淺談
golang中也實(shí)現(xiàn)了排序算法的包sort包,sort 包 在內(nèi)部實(shí)現(xiàn)了四種基本的排序算法:插入排序(insertionSort)、歸并排序(symMerge)、堆排序(heapSort)和快速排序(quickSort); sort 包會(huì)依據(jù)實(shí)際數(shù)據(jù)自動(dòng)選擇最優(yōu)的排序算法。
所以我們寫代碼時(shí)只需要考慮實(shí)現(xiàn) sort.Interface
這個(gè)類型就可以了。
粗略的看看sort包
func Sort(data Interface) { // Switch to heapsort if depth of 2*ceil(lg(n+1)) is reached. n := data.Len() maxDepth := 0 for i := n; i > 0; i >>= 1 { maxDepth++ } maxDepth *= 2 quickSort(data, 0, n, maxDepth) } 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) } // 內(nèi)部實(shí)現(xiàn)的四種排序算法 // 插入排序 func insertionSort(data Interface, a, b int) // 堆排序 func heapSort(data Interface, a, b int) // 快速排序 func quickSort(data Interface, a, b, maxDepth int) // 歸并排序 func symMerge(data Interface, a, m, b int)
所以要調(diào)用sort.Sort()
來實(shí)現(xiàn)自定義類型排序,只需要我們的類型實(shí)現(xiàn) Interface 接口類型中的三個(gè)方法即可。
先看看 sort 包本身對(duì)于 []int
類型如何排序
// 首先定義了一個(gè)[]int類型的別名IntSlice type IntSlice []int // 獲取此 slice 的長度 func (p IntSlice) Len() int { return len(p) } // 比較兩個(gè)元素大小 升序 func (p IntSlice) Less(i, j int) bool { return p[i] < p[j] } // 交換數(shù)據(jù) func (p IntSlice) Swap(i, j int) { p[i], p[j] = p[j], p[i] } // sort.Ints()內(nèi)部調(diào)用Sort() 方法實(shí)現(xiàn)排序 // 注意 要先將[]int 轉(zhuǎn)換為 IntSlice類型 因?yàn)榇祟愋筒艑?shí)現(xiàn)了Interface的三個(gè)方法 func Ints(a []int) { Sort(IntSlice(a)) }
照葫蘆畫瓢 我們來對(duì)自定義的結(jié)構(gòu)體類型進(jìn)行降序排序
package main import ( "fmt" "sort" ) type Person struct { Name string Age int } type Persons []Person // 獲取此 slice 的長度 func (p Persons) Len() int { return len(p) } // 根據(jù)元素的年齡降序排序 (此處按照自己的業(yè)務(wù)邏輯寫) func (p Persons) Less(i, j int) bool { return p[i].Age > p[j].Age } // 交換數(shù)據(jù) func (p Persons) Swap(i, j int) { p[i], p[j] = p[j], p[i] } func main() { persons := Persons{ { Name: "test1", Age: 20, }, { Name: "test2", Age: 22, }, { Name: "test3", Age: 21, }, } fmt.Println("排序前") for _, person := range persons { fmt.Println(person.Name, ":", person.Age) } sort.Sort(persons) fmt.Println("排序后") for _, person := range persons { fmt.Println(person.Name, ":", person.Age) } }
其實(shí),一般 Len()
和 Swap()
基本不做改變,只有涉及到元素比較的 Less()
方法會(huì)有所改變。
當(dāng)我們對(duì)某一個(gè)結(jié)構(gòu)體中多個(gè)字段進(jìn)行排序時(shí)怎么辦,難道每排序一個(gè)就寫下這三個(gè)方法么,當(dāng)然不是。我們可以利用嵌套結(jié)構(gòu)體來解決這個(gè)問題。因?yàn)榍短捉Y(jié)構(gòu)體可以繼承父結(jié)構(gòu)體的所有屬性和方法
比如我想對(duì)上面 Person 的 Name 字段和 Age 對(duì)要排序,我們可以利用嵌套結(jié)構(gòu)體來改進(jìn)一下。
package main import ( "fmt" "sort" ) type Person struct { Name string Age int } type Persons []Person // Len()方法和Swap()方法不用變化 // 獲取此 slice 的長度 func (p Persons) Len() int { return len(p) } // 交換數(shù)據(jù) func (p Persons) Swap(i, j int) { p[i], p[j] = p[j], p[i] } // 嵌套結(jié)構(gòu)體 將繼承 Person 的所有屬性和方法 // 所以相當(dāng)于SortByName 也實(shí)現(xiàn)了 Len() 和 Swap() 方法 type SortByName struct{ Persons } // 根據(jù)元素的姓名長度降序排序 (此處按照自己的業(yè)務(wù)邏輯寫) func (p SortByName) Less(i, j int) bool { return len(p.Persons[i].Name) > len(p.Persons[j].Name) } type SortByAge struct{ Persons } // 根據(jù)元素的年齡降序排序 (此處按照自己的業(yè)務(wù)邏輯寫) func (p SortByAge) Less(i, j int) bool { return p.Persons[i].Age > p.Persons[j].Age } func main() { persons := Persons{ { Name: "test123", Age: 20, }, { Name: "test1", Age: 22, }, { Name: "test12", Age: 21, }, } fmt.Println("排序前") for _, person := range persons { fmt.Println(person.Name, ":", person.Age) } sort.Sort(SortByName{persons}) fmt.Println("排序后") for _, person := range persons { fmt.Println(person.Name, ":", person.Age) } }
總結(jié)
以上就是這篇文章的全部內(nèi)容了,希望本文的內(nèi)容對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,如果有疑問大家可以留言交流,謝謝大家對(duì)腳本之家的支持。
相關(guān)文章
Golang實(shí)現(xiàn)協(xié)程超時(shí)控制的方式總結(jié)
我們知道,go協(xié)程如果不做好處理,很容易造成內(nèi)存泄漏,所以對(duì)goroutine做超時(shí)控制,才能有效避免這種情況發(fā)生,本文為大家整理了兩個(gè)常見的Golang超時(shí)控制方法,需要的可以收藏一下2023-05-05Go語言實(shí)現(xiàn)登錄驗(yàn)證代碼案例
這篇文章主要介紹了Go語言實(shí)現(xiàn)登錄驗(yàn)證代碼案例,代碼和圖文講解的很清晰,有感興趣的可以學(xué)習(xí)下2021-03-03go語言編程實(shí)現(xiàn)遞歸函數(shù)示例詳解
這篇文章主要為大家介紹了go語言編程實(shí)現(xiàn)遞歸函數(shù)示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-09-09