浅谈Go Slice 高级实践
Go 语言切片是对数组的抽象。
Go 数组的长度不可改变,在特定场景中这样的集合就不太适用,Go中提供了一种灵活,功能强悍的内置类型切片("动态数组"),与数组相比切片的长度是不固定的,可以追加元素,在追加时可能使切片的容量增大。
定义切片
你可以声明一个未指定大小的数组来定义切片:
var identifier []type
切片不需要说明长度。
或使用make()函数来创建切片:
var slice1 []type = make([]type, len)
也可以简写为
slice1 := make([]type, len)
也可以指定容量,其中capacity为可选参数。
make([]T, length, capacity)
这里 len 是数组的长度并且也是切片的初始长度。
以下用法中,类型均使用 int64 做为示例,不处理 interface 。
代码只是展示实现思路,不一定完善。
合并两个有序切片,新切片仍然有序
func MergeSortedSlice(s1, s2 []int64) []int64 { // 从末尾元素开始遍历 i := len(s1) - 1 j := len(s2) - 1 // 合并后的长度 newLen := len(s1) + len(s2) // 合并后的索引,也从末尾元素开始 newIdx := newLen - 1 // 创建一个新切片,代表合并后的 newS := make([]int64, newLen) // 将 s1 的内容拷贝到新切片 for k, v := range s1 { newS[k] = v } // 开始遍历 for i >= 0 && j >= 0 { // 新元素 var newNum int64 // 将较大的值赋给新元素,同时向前移动指针 if newS[i] > s2[j] { newNum = newS[i] i-- } else { newNum = s2[j] j-- } newS[newIdx] = newNum newIdx-- } // 如果 s2 还有剩余元素,则剩余元素一定都是最小的,直接放到头部即可 for j >= 0 { newS[newIdx] = s2[j] j-- newIdx-- } return newS}
根据特定规则过滤元素
func FilterSlice(s []int64, filter func(x int64) bool) []int64 { // 返回的新切片 // s[:0] 这种写法是创建了一个 len 为 0,cap 为 len(s) 即和原始切片最大容量一致的切片 // 因为是过滤,所以新切片的元素总个数一定不大于比原始切片,这样做减少了切片扩容带来的影响 // 同时,也有一个问题,因为 newS 和 s 共享底层数组,那么过滤后 s 也会被修改! newS := s[:0] // 遍历,对每个元素执行 filter,符合条件的加入新切片中 for _, x := range s { if !filter(x) { newS = append(newS, x) } } return newS}
去重
两种思路,循环顺序查找和使用 map 加快查找(引入一个 map 在各方面也是有开销的)。选用哪种,可以通过具体场景的 Benchmark 决定
func RemoveDuplicates(s []int64) []int64 { var ret []int64 for _, v := range s { found := false for _, v2 := range ret { if v == v2 { found = true break } } if !found { ret = append(ret, v) } } return ret}func RemoveDuplicates2(s []int64) []int64 { ret := s[:0] // 利用 struct{}{} 减少内存占用 assist := map[int64]struct{}{} for _, v := range s { if _, ok := assist[v]; !ok { assist[v] = struct{}{} ret = append(ret, v) } } return ret}
反转
func Reversing(s []int64) []int64 { for left, right := 0, len(s)-1; left < right; left, right = left+1, right-1 { s[left], s[right] = s[right], s[left] } return s}
分块
主要用于当单个切片过大,需要分多次使用的时候,比如网络调用等。
func SliceChunk(s []int64, size int) [][]int64 { var ret [][]int64 for size < len(s) { // s[:size:size] 表示 len 为 size,cap 也为 size,第二个冒号后的 size 表示 cap s, ret = s[size:], append(ret, s[:size:size]) } ret = append(ret, s) return ret}
类型转换
RPC 中,不同下游接收的类型可能不一样,还有自定义类型,这里提供一个快速转换的方法
s := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}var newS []int64// 做法是利用 reflect 直接替换数据指针// 但是这个不保证在以后的版本中一直可用 r(st)q*(*reflect.SliceHeader)(unsafe.Pointer(&newS)) = *(*reflect.SliceHeader)(unsafe.Pointer(&s))fmt.Printf("type:%T value:%v", newS, newS)
主要参考:https://github.com/golang/go/wiki/SliceTricks
官方使用技巧,建议多看看。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
相关文章