写点什么

切片的其他妙用

作者:宇宙之一粟
  • 2023-04-29
    广东
  • 本文字数:1288 字

    阅读完需:约 4 分钟

切片的其他妙用

过滤而不分配

这个技巧利用了一个切片与原始切片共享相同的支持数组和容量这一事实,因此存储被重新用于过滤后的切片。当然,原始内容是修改过的。

b := a[:0]for _, x := range a {	if f(x) {		b = append(b, x)	}}
复制代码

对于必须进行垃圾回收的元素,可以在之后包含以下代码:

for i := len(b); i < len(a); i++ {	a[i] = nil // or the zero value of T}
复制代码


反转

要用相同的元素但以相反的顺序替换切片的内容:

for i := len(a)/2-1; i >= 0; i-- {	opp := len(a)-1-i	a[i], a[opp] = a[opp], a[i]}
复制代码

同样的事情,除了两个索引:

for left, right := 0, len(a)-1; left < right; left, right = left+1, right-1 {	a[left], a[right] = a[right], a[left]}
复制代码

洗牌

Fisher–Yates 算法:

Since go1.10, this is available at math/rand.Shuffle

for i := len(a) - 1; i > 0; i-- {    j := rand.Intn(i + 1)    a[i], a[j] = a[j], a[i]}
复制代码

最小分配的批处理

如果你想对大切片进行批处理,这很有用。

actions := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}batchSize := 3batches := make([][]int, 0, (len(actions) + batchSize - 1) / batchSize)
for batchSize < len(actions) { actions, batches = actions[batchSize:], append(batches, actions[0:batchSize:batchSize])}batches = append(batches, actions)
复制代码

产生以下结果:

[[0 1 2] [3 4 5] [6 7 8] [9]]
复制代码

就地重复数据删除(比较)

import "sort"
in := []int{3,2,1,4,3,2,1,4,1} // any item can be sortedsort.Ints(in)j := 0for i := 1; i < len(in); i++ { if in[j] == in[i] { continue } j++ // preserve the original data // in[i], in[j] = in[j], in[i] // only set what is required in[j] = in[i]}result := in[:j+1]fmt.Println(result) // [1 2 3 4]
复制代码


如果可能的话,移动到前面,或者如果不存在则放在前面。

// moveToFront moves needle to the front of haystack, in place if possible.func moveToFront(needle string, haystack []string) []string {	if len(haystack) != 0 && haystack[0] == needle {		return haystack	}	prev := needle	for i, elem := range haystack {		switch {		case i == 0:			haystack[0] = needle			prev = elem		case elem == needle:			haystack[i] = prev			return haystack		default:			haystack[i] = prev			prev = elem		}	}	return append(haystack, prev)}
haystack := []string{"a", "b", "c", "d", "e"} // [a b c d e]haystack = moveToFront("c", haystack) // [c a b d e]haystack = moveToFront("f", haystack) // [f c a b d e]
复制代码

滑动窗口

func slidingWindow(size int, input []int) [][]int {	// returns the input slice as the first element	if len(input) <= size {		return [][]int{input}	}
// allocate slice at the precise size we need r := make([][]int, 0, len(input)-size+1)
for i, j := 0, size; j <= len(input); i, j = i+1, j+1 { r = append(r, input[i:j]) }
return r}
复制代码


发布于: 刚刚阅读数: 2
用户头像

宇宙古今无有穷期,一生不过须臾,当思奋争 2020-05-07 加入

🏆InfoQ写作平台-签约作者 🏆 混迹于江湖,江湖却没有我的影子 热爱技术,专注于后端全栈,轻易不换岗 拒绝内卷,工作于外企开发,弹性不加班 热衷分享,执着于阅读写作,佛系不水文 同名公众号:《宇宙之一粟》

评论

发布
暂无评论
切片的其他妙用_Go_宇宙之一粟_InfoQ写作社区