Go排序

Posted 2022-07-18 nyist-xsk

本文链接：https://blog.csdn.net/u011304970/article/details/71447148

简介
Go的sort包提供了排序功能。包括基本类型的排序和自定义数据（通常是结构体数组）的排序。

基本类型的排序
sort提供了以下API对基本类型进行排序，查找

// 排序
func Ints(a []int)
func Float64s(a []float64)
func Strings(a []string)

// 判断是否已经排序
func IntsAreSorted(a []int) bool
func Float64sAreSorted(a []float64) bool
func StringsAreSorted(a []string) bool

// 在已排序的slice中查找，返回索引值，如果找不到，返回len(slice)
func SearchStrings(a []string, x string) int
func SearchFloat64s(a []float64, x float64) int
func SearchInts(a []int, x int) int

package main

import (
    "fmt"
    "sort"
)

func main() 
    is := []int3, 5, 2
    fmt.Println(sort.IntsAreSorted(is)) // false
    sort.Ints(is)
    fmt.Println(is)                     // [2 3 5]
    fmt.Println(sort.IntsAreSorted(is)) // true
    fmt.Println(sort.SearchInts(is, 5)) // 2
    fmt.Println(sort.SearchInts(is, 8)) // 3

自定义类型的排序
自定义类型的排序分为两种，一种是通过传递函数值，一种是通过实现接口

传递函数值

// 对slice排序
func Slice(slice interface, less func(i, j int) bool)
func SliceStable(slice interface, less func(i, j int) bool) // 稳定排序
// 判断是否已经排序
func SliceIsSorted(slice interface, less func(i, j int) bool) bool
// 二分法查找，这个函数很有意思，它查找并返回使函数f返回true和返回false的临界点（见下例）
func Search(n int, f func(int) bool) int

Slice()和SliceIsSorted()用法

package main

import (
    "fmt"
    "sort"
)

type Person struct 
    Name string
    Age  int


func main() 

    persons := []Person
        Person"Gopher", 11,
        Person"Monkey", 12,
        Person"Cynhard", 5,
    

    // 按照名字排序
    less := func(i, j int) bool 
        return persons[i].Name < persons[j].Name
    

    fmt.Println(sort.SliceIsSorted(persons, less)) // false
    sort.Slice(persons, less)
    fmt.Println(persons)                           // [Cynhard 5 Gopher 11 Monkey 12]
    fmt.Println(sort.SliceIsSorted(persons, less)) // true

Search()用法

package main

import (
    "fmt"
    "sort"
)

func main() 

    ints := []int22, 34, 21, 32, 54, 64, 49, 43
    sort.Ints(ints)
    fmt.Println(ints) // [21 22 32 34 43 49 54 64]

    // ints[0]~ints[1]都小于32，ints[2]~ints[7]都大于等于32
    // 因此临界点索引为2，found==2
    found := sort.Search(len(ints), func(i int) bool 
        return ints[i] >= 32
    )
    fmt.Println(found) // 2
    if found < len(ints) && ints[found] == 32 
        fmt.Printf("32 found at index %d\n", found)
     else  // 没有找到32，但是返回了32应该插入的位置
        fmt.Println("32 not found")
    

    // 由于找不到一个临界点，使序列左边为32，右边不为32
    // 所以返回len(ints)，found==8
    found = sort.Search(len(ints), func(i int) bool 
        return ints[i] == 32
    )
    fmt.Println(found) // 8

实现接口
sort提供了一个接口：sort.Interface，所有满足这个接口的自定义类型都可以进行排序。API如下

type Interface interface 
        Len() int
        Less(i, j int) bool
        Swap(i, j int)

func Sort(data Interface)
func IsSorted(data Interface) bool
func Stable(data Interface)
// 返回data的逆序接口
func Reverse(data Interface) Interface

例如：

package main

import (
    "fmt"
    "sort"
)

type Person struct 
    Name string
    Age  int


type PerColl []Person

var persons PerColl

func (PerColl) Len() int 
    return len(persons)


func (PerColl) Less(i, j int) bool 
    return persons[i].Name < persons[j].Name


func (PerColl) Swap(i, j int) 
    persons[i], persons[j] = persons[j], persons[i]


func main() 

    persons = []Person
        Person"Cynhard", 5,
        Person"Gopher", 11,
        Person"Monkey", 12,
    

    sort.Sort(persons)
    fmt.Println(persons)                // [Cynhard 5 Gopher 11 Monkey 12]
    fmt.Println(sort.IsSorted(persons)) // true
    sort.Sort(sort.Reverse(persons))    
    fmt.Println(persons)                // [Monkey 12 Gopher 11 Cynhard 5]