Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Go1.21 slices包

Go1.21新增slices包的用法详解

作者:路多辛

Go 1.21新增的 slices 包提供了很多和切片相关的函数,可以用于任何类型的切片,这篇文章主要来和大家介绍一下slices包中相关函数的用法,需要的可以参考一下

slices.BinarySearch

定义如下:

func BinarySearch[S ~[]E, E cmp.Ordered](x S, target E) (int, bool)

在已经排好序的切片(切片必须按递增顺序排序)中搜索目标,如果找到了,返回所在的位置和 true,如果没有找到,则返回目标应该被找到的位置和 false,简单示例如下:

package main
import (
    "fmt"
    "slices"
)
func main() {
    names := []string{"Alice", "Bob", "Vera"}
    n, found := slices.BinarySearch(names, "Vera")
    fmt.Println("Vera:", n, found) // Vera: 2 true
    n, found = slices.BinarySearch(names, "Bill")
    fmt.Println("Bill:", n, found) // Bill: 1 false
}

slices.BinarySearchFunc

定义如下:

func BinarySearchFunc[S ~[]E, E, T any](x S, target T, cmp func(E, T) int) (int, bool)

作用类似 slices.BinarySearch,不同的是使用自定义比较函数。切片必须按递增顺序排序,其中“递增”是由cmp定义的。如果 slice 元素与目标匹配,CMP应该返回 0,如果 slice元素在目标之前,则返回一个负数,如果 slice 元素在目标之后,则返回一个正数。CMP必须实现与片相同的排序,这样,如果CMP (a, t) < 0且CMP (b, t) >= 0,则切中的 a 必须位于 b 之前。

package main
import (
    "cmp"
    "fmt"
    "slices"
)
func main() {
    type Person struct {
        Name string
        Age  int
    }
    people := []Person{
        {"Alice", 55},
        {"Bob", 24},
        {"Gopher", 13},
    }
    n, found := slices.BinarySearchFunc(people, Person{"Bob", 0}, func(a, b Person) int {
        return cmp.Compare(a.Name, b.Name)
    })
    fmt.Println("Bob:", n, found) // Bob: 1 true
}

slices.Clip

定义如下:

func Clip[S ~[]E, E any](s S) S

从切片中删除未使用的容量,返回 s[:len(s):len(s)]。简单示例如下:

package main
import (
    "fmt"
    "slices"
)
func main() {
    names := make([]string, 2, 5)
    names = slices.Clip(names)
    fmt.Printf("长度:%d,容量:%d", len(names), cap(names))
  // 长度:2,容量:2
}

slices.Clone

定义如下:

func Clone[S ~[]E, E any](s S) S

返回切片的副本。因为元素是使用赋值方式复制的,所以这是一个浅克隆。简单实用方法如下:

package main
import (
    "fmt"
    "slices"
)
func main() {
    names := []string{"路多辛的博客", "路多辛的所思所想"}
    namesCopy := slices.Clone(names)
    fmt.Println(namesCopy)
}

slices.Compact

定义如下:

func Compact[S ~[]E, E comparable](s S) S

将连续出现的元素变为一个,类似于 Unix 上的 uniq 命令。Compact 会修改片的内容并返回修改后的片,长度可能会变得更小。简单示例如下:

package main
import (
    "fmt"
    "slices"
)
func main() {
    seq := []int{0, 1, 1, 2, 5, 5, 5, 8}
    seq = slices.Compact(seq)
    fmt.Println(seq) // [0 1 2 5 8]
}

slices.CompactFunc

定义如下:

func CompactFunc[S ~[]E, E any](s S, eq func(E, E) bool) S

类似于 slices.Compact,不同的是使用自定义的函数来比较元素。如果元素的运行结果相等,CompactFunc 将保留第一个元素。简单示例如下:

package main
import (
    "fmt"
    "slices"
    "strings"
)
func main() {
    names := []string{"bob", "Bob", "alice", "Vera", "VERA"}
    names = slices.CompactFunc(names, func(a, b string) bool {
        return strings.ToLower(a) == strings.ToLower(b)
    })
    fmt.Println(names) // [bob alice Vera]
}

slices.Compare

定义如下:

func Compare[S ~[]E, E cmp.Ordered](s1, s2 S) int

使用 cmp.Compare函数来比较 s1 和 s2 的元素。按照顺序比较每一对元素,直到一个元素不等于另一个元素。返回第一个不匹配元素的结果。如果两个切片在其中一个结束之前相等,则认为较短的切片小于较长的切片。如果 s1 == s2,结果为0;如果 s1 < s2,结果为-1;如果 s1 > s2,结果为1。

package main
import (
    "fmt"
    "slices"
)
func main() {
    names := []string{"Alice", "Bob", "Vera"}
    fmt.Println("Equal:", slices.Compare(names, []string{"Alice", "Bob", "Vera"}))
    fmt.Println("V < X:", slices.Compare(names, []string{"Alice", "Bob", "Xena"}))
    fmt.Println("V > C:", slices.Compare(names, []string{"Alice", "Bob", "Cat"}))
    fmt.Println("3 > 2:", slices.Compare(names, []string{"Alice", "Bob"}))
}

运行结果如下:

Equal: 0
V < X: -1
V > C: 1
3 > 2: 1

slices.CompareFunc

定义如下:

func CompareFunc[S1 ~[]E1, S2 ~[]E2, E1, E2 any](s1 S1, s2 S2, cmp func(E1, E2) int) int

类似于 slices.Compare,不同的是使用自定义的比较函数进行比较。结果是 cmp 的第一个非零结果;如果 cmp 总是返回0,则如果 len(s1) == len(s2) 结果为0,如果len(s1) < len(s2)结果为-1,如果 len(s1) > len(s2) 结果为1。简单示例如下:

package main
import (
    "cmp"
    "fmt"
    "slices"
    "strconv"
)
func main() {
    numbers := []int{0, 43, 8}
    strings := []string{"0", "0", "8"}
    result := slices.CompareFunc(numbers, strings, func(n int, s string) int {
        sn, err := strconv.Atoi(s)
        if err != nil {
            return 1
        }
        return cmp.Compare(n, sn)
    })
    fmt.Println(result) // 1
}

slices.Contains

定义如下:

func Contains[S ~[]E, E comparable](s S, v E) bool

用于判断 s 是否包含 v。简单示例如下:

package main
import (
    "fmt"
    "slices"
)
func main() {
    names := []string{"Alice", "Bob", "Vera"}
    fmt.Println(slices.Contains(names, "Bob")) // true
}

slices.ContainsFunc

定义如下:

func ContainsFunc[S ~[]E, E any](s S, f func(E) bool) bool

用于判断 s 中是否至少有一个元素 e 满足 f(e)。简单示例如下:

package main
import (
    "fmt"
    "slices"
)
func main() {
    numbers := []int{0, 42, -10, 8}
    hasNegative := slices.ContainsFunc(numbers, func(n int) bool {
        return n < 0
    })
    fmt.Println("Has a negative:", hasNegative)// true
    hasOdd := slices.ContainsFunc(numbers, func(n int) bool {
        return n%2 != 0
    })
    fmt.Println("Has an odd number:", hasOdd) // false
}

【参考资料】

Package slices(https://golang.google.cn/pkg/slices/

以上就是Go1.21新增slices包的用法详解的详细内容,更多关于Go1.21 slices包的资料请关注脚本之家其它相关文章!

您可能感兴趣的文章:
阅读全文