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 }