本文链接:https://blog.csdn.net/u011304970/article/details/71447148
简介
Go的sort包提供了排序功能。包括基本类型的排序和自定义数据(通常是结构体数组)的排序。
基本类型的排序
sort提供了以下API对基本类型进行排序,查找
// 排序
func Ints(a []int)
func Float64s(a []float64)
func Strings(a []string)
// 判断是否已经排序
func IntsAreSorted(a []int) bool
func Float64sAreSorted(a []float64) bool
func StringsAreSorted(a []string) bool
// 在已排序的slice中查找,返回索引值,如果找不到,返回len(slice)
func SearchStrings(a []string, x string) int
func SearchFloat64s(a []float64, x float64) int
func SearchInts(a []int, x int) int
package main
import (
"fmt"
"sort"
)
func main() {
is := []int{3, 5, 2}
fmt.Println(sort.IntsAreSorted(is)) // false
sort.Ints(is)
fmt.Println(is) // [2 3 5]
fmt.Println(sort.IntsAreSorted(is)) // true
fmt.Println(sort.SearchInts(is, 5)) // 2
fmt.Println(sort.SearchInts(is, 8)) // 3
}
自定义类型的排序
自定义类型的排序分为两种,一种是通过传递函数值,一种是通过实现接口
传递函数值
// 对slice排序
func Slice(slice interface{}, less func(i, j int) bool)
func SliceStable(slice interface{}, less func(i, j int) bool) // 稳定排序
// 判断是否已经排序
func SliceIsSorted(slice interface{}, less func(i, j int) bool) bool
// 二分法查找,这个函数很有意思,它查找并返回使函数f返回true和返回false的临界点(见下例)
func Search(n int, f func(int) bool) int
Slice()和SliceIsSorted()用法
package main
import (
"fmt"
"sort"
)
type Person struct {
Name string
Age int
}
func main() {
persons := []Person{
Person{"Gopher", 11},
Person{"Monkey", 12},
Person{"Cynhard", 5},
}
// 按照名字排序
less := func(i, j int) bool {
return persons[i].Name < persons[j].Name
}
fmt.Println(sort.SliceIsSorted(persons, less)) // false
sort.Slice(persons, less)
fmt.Println(persons) // [{Cynhard 5} {Gopher 11} {Monkey 12}]
fmt.Println(sort.SliceIsSorted(persons, less)) // true
}
Search()用法
package main
import (
"fmt"
"sort"
)
func main() {
ints := []int{22, 34, 21, 32, 54, 64, 49, 43}
sort.Ints(ints)
fmt.Println(ints) // [21 22 32 34 43 49 54 64]
// ints[0]~ints[1]都小于32,ints[2]~ints[7]都大于等于32
// 因此临界点索引为2,found==2
found := sort.Search(len(ints), func(i int) bool {
return ints[i] >= 32
})
fmt.Println(found) // 2
if found < len(ints) && ints[found] == 32 {
fmt.Printf("32 found at index %d\n", found)
} else { // 没有找到32,但是返回了32应该插入的位置
fmt.Println("32 not found")
}
// 由于找不到一个临界点,使序列左边为32,右边不为32
// 所以返回len(ints),found==8
found = sort.Search(len(ints), func(i int) bool {
return ints[i] == 32
})
fmt.Println(found) // 8
}
实现接口
sort提供了一个接口:sort.Interface,所有满足这个接口的自定义类型都可以进行排序。API如下
type Interface interface {
Len() int
Less(i, j int) bool
Swap(i, j int)
}
func Sort(data Interface)
func IsSorted(data Interface) bool
func Stable(data Interface)
// 返回data的逆序接口
func Reverse(data Interface) Interface
例如:
package main
import (
"fmt"
"sort"
)
type Person struct {
Name string
Age int
}
type PerColl []Person
var persons PerColl
func (PerColl) Len() int {
return len(persons)
}
func (PerColl) Less(i, j int) bool {
return persons[i].Name < persons[j].Name
}
func (PerColl) Swap(i, j int) {
persons[i], persons[j] = persons[j], persons[i]
}
func main() {
persons = []Person{
Person{"Cynhard", 5},
Person{"Gopher", 11},
Person{"Monkey", 12},
}
sort.Sort(persons)
fmt.Println(persons) // [{Cynhard 5} {Gopher 11} {Monkey 12}]
fmt.Println(sort.IsSorted(persons)) // true
sort.Sort(sort.Reverse(persons))
fmt.Println(persons) // [{Monkey 12} {Gopher 11} {Cynhard 5}]
}
|
请发表评论