golang中怎么对自定义类型进行排序
这篇文章给大家介绍golang中怎么对自定义类型进行排序,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。
创新互联公司 - 成都IDC机房托管,四川服务器租用,成都服务器租用,四川网通托管,绵阳服务器托管,德阳服务器托管,遂宁服务器托管,绵阳服务器托管,四川云主机,成都云主机,西南云主机,成都IDC机房托管,西南服务器托管,四川/成都大带宽,大带宽服务器,四川老牌IDC服务商
sort包浅谈
golang中也实现了排序算法的包sort包,sort 包 在内部实现了四种基本的排序算法:插入排序(insertionSort)、归并排序(symMerge)、堆排序(heapSort)和快速排序(quickSort); sort 包会依据实际数据自动选择最优的排序算法。
所以我们写代码时只需要考虑实现 sort.Interface
这个类型就可以了。
粗略的看看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) } // 内部实现的四种排序算法 // 插入排序 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)
所以要调用sort.Sort()
来实现自定义类型排序,只需要我们的类型实现 Interface 接口类型中的三个方法即可。
先看看 sort 包本身对于 []int
类型如何排序
// 首先定义了一个[]int类型的别名IntSlice type IntSlice []int // 获取此 slice 的长度 func (p IntSlice) Len() int { return len(p) } // 比较两个元素大小 升序 func (p IntSlice) Less(i, j int) bool { return p[i] < p[j] } // 交换数据 func (p IntSlice) Swap(i, j int) { p[i], p[j] = p[j], p[i] } // sort.Ints()内部调用Sort() 方法实现排序 // 注意 要先将[]int 转换为 IntSlice类型 因为此类型才实现了Interface的三个方法 func Ints(a []int) { Sort(IntSlice(a)) }
照葫芦画瓢 我们来对自定义的结构体类型进行降序排序
package main import ( "fmt" "sort" ) type Person struct { Name string Age int } type Persons []Person // 获取此 slice 的长度 func (p Persons) Len() int { return len(p) } // 根据元素的年龄降序排序 (此处按照自己的业务逻辑写) func (p Persons) Less(i, j int) bool { return p[i].Age > p[j].Age } // 交换数据 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) } }
其实,一般 Len()
和 Swap()
基本不做改变,只有涉及到元素比较的 Less()
方法会有所改变。
当我们对某一个结构体中多个字段进行排序时怎么办,难道每排序一个就写下这三个方法么,当然不是。我们可以利用嵌套结构体来解决这个问题。因为嵌套结构体可以继承父结构体的所有属性和方法
比如我想对上面 Person 的 Name 字段和 Age 对要排序,我们可以利用嵌套结构体来改进一下。
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) } // 交换数据 func (p Persons) Swap(i, j int) { p[i], p[j] = p[j], p[i] } // 嵌套结构体 将继承 Person 的所有属性和方法 // 所以相当于SortByName 也实现了 Len() 和 Swap() 方法 type SortByName struct{ Persons } // 根据元素的姓名长度降序排序 (此处按照自己的业务逻辑写) func (p SortByName) Less(i, j int) bool { return len(p.Persons[i].Name) > len(p.Persons[j].Name) } type SortByAge struct{ Persons } // 根据元素的年龄降序排序 (此处按照自己的业务逻辑写) 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) } }
关于golang中怎么对自定义类型进行排序就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。
网站栏目:golang中怎么对自定义类型进行排序
网站URL:http://ybzwz.com/article/iphchc.html