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

原文链接：http://www.zhoubotong.site/post/44.html

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

    以上就介绍这么多了。感兴趣的朋友欢迎留言补充哈。