sort包簡介 官方文檔 Golang的sort包用來排序，二分查找等操作。本文主要介紹sort包里常用的函數，通過實例代碼來快速學會使用sort包 sort包內置函數 sort.Ints(x []int) ints := []int{1, 4, 3, 2} fmt.Printf("%v\n", i ...
sort包簡介
官方文檔
Golang的sort包用來排序，二分查找等操作。本文主要介紹sort包里常用的函數，通過實例代碼來快速學會使用sort包
sort包內置函數
sort.Ints(x []int)
ints := []int{1, 4, 3, 2}
fmt.Printf("%v\n", ints)
sort.Ints(ints) //預設升序
fmt.Printf("%v\n", ints) //[1 2 3 4]
sort.Sort(sort.Reverse(sort.IntSlice(ints))) //降序排序
fmt.Printf("%v\n", ints) //[4 3 2 1]
sort.Strings(x []string) sort.Float64s(x []float64)
- 使用方法同上，都是對內置
int string float64類型的便捷排序
sort.Slice(x any, less func(i, j int) bool)
- 傳入對象是切片，要自己實現回調函數
slices := []int{1, 1, 4, 5, 1, 4}
sort.Slice(slices, func(i, j int) bool {
return slices[i] < slices[j]
})
fmt.Printf("%v\n", slices)//[1 1 1 4 4 5]
- 同時也可以對結構體自定義排序規則
type stu struct {
name string
age int
}
stus := []stu{{"h", 20}, {"a", 23}, {"h", 21}}
sort.Slice(stus, func(i, j int) bool {
if stus[i].name == stus[j].name {
return stus[i].age > stus[j].age // 年齡逆序
}
return stus[i].name < stus[j].name // 名字正序
})
fmt.Printf("%v\n", stus) //[{a 23} {h 21} {h 20}]
sort.Sort(data Interface)
- 自定義排序，需要實現
Len() Less() Swap()三個方法
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)
}
- 使用代碼
type stu struct {
name string
age int
}
type student []stu
func (s student) Len() int {
return len(s)
}
func (s student) Less(i, j int) bool {
if s[i].name == s[j].name {
return s[i].age > s[j].age // 年齡逆序
}
return s[i].name < s[j].name // 名字正序
}
func (s student) Swap(i, j int) {
s[i], s[j] = s[j], s[i]
}
func main() {
stus1 := student{{"h", 20}, {"a", 23}, {"h", 21}}
sort.Sort(stus1)
fmt.Printf("%v\n", stus1) //[{a 23} {h 21} {h 20}] 使用效果等同於sort.Slice
}
- 使用效果等同於
sort.Slice後者代碼量較少
sort.SearchInts(a []int, x int) int
- 該函數是用來二分查找的, 預設是在左邊插入
arr := []int{1, 2, 3, 4, 5, 6, 7}
idx := sort.SearchInts(arr, 4)
fmt.Printf("%v\n", idx) // 3
sort.SearchFloat64s(a []float64, x float64) int sort.SearchStrings(a []string, x string) int
- 這兩函數功能同上
sort.Search(n int, f func(int) bool) int
- 自定義的二分查找,回調函數需要自己實現查找條件
arr := []int{1, 2, 3, 4, 5, 6, 7}
idx := sort.Search(len(arr), func(i int) bool {
return arr[i] > 4
})
fmt.Printf("%v\n", idx) //4
- 相比
SearchInts,通過自定義條件便實現了相等情況下在右邊插入，前者預設是在左邊 - 更高級一點的用法
mysring := []string{"abcd", "bcde", "bfag", "cddd"}
idx := sort.Search(len(mysring), func(i int) bool {
// 查找頭兩位字母不是b的，，返回找到的第一個
return mysring[i][0] != 'b' && mysring[i][1] != 'b'
})
fmt.Printf("%v\n", mysring[idx]) // cddd
mysring := []string{"abcd", "bcde", "bfag", "cddd"}
idx := sort.Search(len(mysring), func(i int) bool {
//查找第一個字母不是b的
return mysring[i][0] <= byte('b')
})
fmt.Printf("%v\n", mysring[idx]) // abcd
