Golang中字符串、数组、切片排序
使用Golang的sort包用来排序,包括二分查找等操作。下面通过实例代码来分享下sort包的使用技巧:
package main
import (
"fmt"
"sort"
)
func main() {
ints := []int{11, 44, 33, 22}
sort.Ints(ints) //默认升序
fmt.Printf("%v\n", ints) //[11 22 33 44]
sort.Sort(sort.Reverse(sort.IntSlice(ints))) //降序排序
fmt.Printf("%v\n", ints) //[44 33 22 11]
//demo-1:使用字符串排序
//sort.Strings(x []string) sort.Float64s(x []float64)
//使用方法同上,都是对内置int string float64类型的便捷排序
str := []string{"apple", "lemen", "banana", "fruit"}
sort.Strings(str)
fmt.Printf("%v\n", str) //默认升序
sort.Sort(sort.Reverse(sort.StringSlice(str))) //降序排序
fmt.Printf("%v\n", str)
//demo-2:使用切片排序,传入对象是切片,要自己实现回调函数
slices := []int{11, 11, 44, 55, 11, 44}
sort.Slice(slices, func(i, j int) bool {
//return slices[i] < slices[j] //升序 即前面的值比后面的小
return slices[i] > slices[j] //降序 即前面的值比后面的大
})
fmt.Printf("%v\n", slices) //[55 44 44 11 11 11]
//demo-3:对结构体自定义排序规则
type student struct {
name string
age int
}
s := []student{{"babala", 52}, {"anly", 50}, {"babala", 51}}
sort.Slice(s, func(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 // 名字升序
})
fmt.Printf("%v\n", s) //[{anly 50} {babala 52} {babala 51}]
}
使用接口排序:
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)
}
完整代码如下:
package main
import (
"fmt"
"sort"
)
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() {
s1 := student{{"babala", 52}, {"anly", 50}, {"babala", 51}}
sort.Sort(s1)
fmt.Printf("%v\n", s1) //[{anly 50} {babala 52} {babala 51}]使用效果等同于sort.Slice
}
上面可以看到,使用效果等同于sort.Slice,只是后者代码量较少。
sort.SearchInts(a []int, x int) int该函数是用来二分查找的, 默认是在左边插入package main import ( "fmt" "sort" ) func main() { arr := []int{11, 22, 33, 44, 55, 66, 77} idx := sort.SearchInts(arr, 44) // 查找44所在的索引位置,默认0开始 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自定义的二分查找,回调函数需要自己实现查找条件
package main import ( "fmt" "sort" ) func main() { arr := []int{11, 22, 33, 44, 55, 66, 77} idx := sort.Search(len(arr), func(i int) bool { return arr[i] > 44 //查找 >44的值得索引位置即第一个满足条件的55所在的索引 }) fmt.Printf("%v\n", idx) //55 索引位置为4 }相比SearchInts,通过自定义条件便实现了相等情况下在右边插入,前者默认是在左边
更高级一点的用法
package main import ( "fmt" "sort" ) func main() { mysring := []string{"abcd", "bcde", "cdef", "dbac"} 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]) // cdef }

