Golang中字符串、数组、切片排序

作者: adm 分类: go 发布时间: 2023-11-28

使用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
}

如果觉得我的文章对您有用,请随意赞赏。您的支持将鼓励我继续创作!