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
    }
    

    以上就介紹這麼多了。感興趣的朋友歡迎留言補充哈。