Go 1.21新增的 slices 包详解(一)

Go 1.21新增的 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 会修改片的内容并返回修改后的片,长度可能会变得更小。简单示例如下:

扫描二维码关注公众号,回复: 16602692 查看本文章
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/)

猜你喜欢

转载自blog.csdn.net/luduoyuan/article/details/132330420