Go 1.21新增的 slices 包提供了很多和切片相关的函数,可以用于任何类型的切片。
定义如下:
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
}
定义如下:
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
}
定义如下:
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
}
定义如下:
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)
}
定义如下:
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]
}
定义如下:
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]
}
定义如下:
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
定义如下:
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
}
定义如下:
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
}
定义如下:
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/)