Golang Slice技巧

Gundy發表於2019-04-10

追加元素

a = append(a, b...)
複製程式碼

複製

b = make([]T, len(a))
copy(b,a)
// or
b = append([]T(nil), a...)
// or
b = append(a[:0:0], a...) 
複製程式碼

裁剪

a = append(a[:i], a[j:]...)
複製程式碼

刪除元素

a = append(a[:i], a[i+1:]...)
// or
a = a[:i+copy(a[i:], a[i+1:])]
複製程式碼

刪除而不保留順序

a[i] = a[len(a)-1] 
a = a[:len(a)-1]
複製程式碼

如果元素的型別是指標或帶指標欄位的結構,需要對其進行垃圾收集,那麼上面的Cut和Delete實現有潛在的記憶體洩漏問題:一些帶值的元素仍然被片a引用,因此無法收集。下面的程式碼可以修復這個問題

裁剪

copy(a[i:], a[j:])
for k, n := len(a)-j+i, len(a); k < n; k++ {
	a[k] = nil // or the zero value of T
}
a = a[:len(a)-j+i]
複製程式碼

刪除

copy(a[i:], a[i+1:])
a[len(a)-1] = nil // or the zero value of T
a = a[:len(a)-1]
複製程式碼

刪除不保留順序

a[i] = a[len(a)-1]
a[len(a)-1] = nil
a = a[:len(a)-1]
複製程式碼

擴大

a = append(a[:i], append(make([]T, j), a[i:]...)...)
複製程式碼

延伸

a = append(a, make([]T, j)...)
複製程式碼

插入

a = append(a[:i], append([]T{x}, a[i:]...)...)
複製程式碼

第二個append建立一個具有自己底層儲存的新的切片,並將a[i:]中的元素複製到該切片,然後將這些元素複製回切片a(通過第一個append)。可以使用另一種方法避免建立新的片(以及記憶體垃圾)和第二個副本

插入

s = append(s, 0 /* use the zero value of the element type */)
copy(s[i+1:], s[i:])
s[i] = x
複製程式碼

插入Slice

a = append(a[:i], append(b, a[i:]...)...)
複製程式碼

增加元素

a = append(a, x)
複製程式碼

取出最後一個元素

x, a = a[len(a)-1], a[:len(a)-1]
複製程式碼

Push Front/Unshift

a = append([]T{x}, a...)
複製程式碼

Pop Front/Shift

x, a = a[0], a[1:]
複製程式碼

其他技巧

不進行記憶體分配過濾

讓一個切片與原始切片共享底層陣列即可

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]
}
複製程式碼

打亂順序

for i := len(a) - 1; i > 0; i-- {
    j := rand.Intn(i + 1)
    a[i], a[j] = a[j], a[i]
}
複製程式碼
words := strings.Fields("ink runs from the corners of my mouth")
rand.Shuffle(len(words), func(i, j int) {
    words[i], words[j] = words[j], words[i]
})
fmt.Println(words)
複製程式碼

批處理

actions := []int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
batchSize := 3
var batches [][]int

for batchSize < len(actions) {
    actions, batches = actions[batchSize:], append(batches, actions[0:batchSize:batchSize])
}
batches = append(batches, actions)

output:
[[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 sorted
sort.Ints(in)
j := 0
for 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]
複製程式碼

翻譯自:github.com/golang/go/w…

相關文章