当前位置：首页 > article >正文

【go从零单排】sorting排序、Sorting by Functions排序方法

article 2024/11/14 10:56:56

挪威特罗姆瑟夜景

🌈Don’t worry , just coding!
内耗与overthinking只会削弱你的精力，虚度你的光阴，每天迈出一小步，回头时发现已经走了很远。

📗概念

在 Go 语言中，排序通常使用 sort 包。这个包提供了对切片和用户自定义类型的排序功能。

💻代码

sorting排序

package main

//导入slices包，提供了对切片的操作函数，包括排序和检查排序状态
import (
	"fmt"
	"slices"
)

func main() {
	//创建字符串类型的切片strs并赋值
	strs := []string{"c", "a", "b"}
	//排序slice.Sort()
	slices.Sort(strs)
	fmt.Println("Strings:", strs)
	//创建一个int类型的切片并赋值
	ints := []int{7, 2, 4}
	//排序slice.Sort()
	slices.Sort(ints)
	fmt.Println("Ints:   ", ints)
	//	slices.IsSorted() 返回是否已排序，已排序true，未排序false
	s := slices.IsSorted(ints)
	fmt.Println("Sorted: ", s)
}

//输出
//Strings: [a b c]
//Ints:    [2 4 7]
//Sorted:  true

Sorting by Functions排序方法

package main

//导入cmp包用来比较值，导入slices操作切片
import (
	"cmp"
	"fmt"
	"slices"
)

func main() {
	//创建string类型的切片fruits
	fruits := []string{"peach", "banana", "kiwi"}
	//定义比较函数，输入两个string，输出一个int
	lenCmp := func(a, b string) int {
		//返回两个string的长度差
		return cmp.Compare(len(a), len(b))
	}
	//使用slices.SortFunc()排序，按照lenCmp排序
	slices.SortFunc(fruits, lenCmp)
	fmt.Println(fruits)
	//定义结构体Person
	type Person struct {
		name string
		age  int
	}
	//创建Person实例并赋值给people切片
	people := []Person{
		Person{name: "Jax", age: 37},
		Person{name: "TJ", age: 25},
		Person{name: "Alex", age: 72},
	}
	//对传入的people排序，按照age排序
	slices.SortFunc(people,
		func(a, b Person) int {
			//使用cmp.Compare比较两个Person对象的年龄
			return cmp.Compare(a.age, b.age)
		})
	fmt.Println(people)
}

//输出
//[kiwi peach banana]
//[{TJ 25} {Jax 37} {Alex 72}]

sort.Interface

package main

import (
    "fmt"
    "sort"
)

type Person struct {
    Name string
    Age  int
}

type ByAge []Person

func (a ByAge) Len() int           { return len(a) }
func (a ByAge) Less(i, j int) bool { return a[i].Age < a[j].Age }
func (a ByAge) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }

func main() {
    people := []Person{
        {"Alice", 30},
        {"Bob", 25},
        {"Charlie", 35},
    }

    sort.Sort(ByAge(people))
    fmt.Println(people) // 输出: [{Bob 25} {Alice 30} {Charlie 35}]
}

🔍理解

对于基本类型（如整数、浮点数和字符串），可以直接使用 sort 包中的 Sort 函数。
对自定义类型或根据特定条件进行排序，可以实现 sort.Interface 接口。这个接口包含三个方法：Len()、Less(i, j int) bool 和 Swap(i, j int)。
使用 sort.Slice 可以更简洁地进行排序，无需定义新的类型。

💡 Tips小知识点

sort.Interface 接口

Len(): 返回集合的长度，用于确定排序的范围。
Less(i, j int) bool: 定义排序的顺序，返回 true 表示索引 i 的元素应该排在索引 j 的元素之前。
Swap(i, j int): 交换索引 i 和 j 的元素。
必须实现三个方法，只实现了其中一个或两个方法，编译器会提示错误。

cmp.compare()

在 Go 语言中，cmp.Compare 是来自 google/go-cmp
包的一个方法，用于比较两个值。这个包提供了一个灵活的方式来比较结构体、切片和其他数据类型，特别是在单元测试中验证结构是否相等时非常有用

Example：

package main

import (
    "fmt"
    "github.com/google/go-cmp/cmp"
)

type Person struct {
    Name string
    Age  int
}

func main() {
    p1 := Person{"Alice", 30}
    p2 := Person{"Bob", 25}

    // 使用 cmp.Compare 比较两个结构体
    result := cmp.Compare(p1, p2)

    switch {
    case result < 0:
        fmt.Println("p1 is less than p2")
    case result > 0:
        fmt.Println("p1 is greater than p2")
    default:
        fmt.Println("p1 is equal to p2")
    }
}