Golang切片刪除指定元素的三種方法對比

更新時間：2022年06月29日 15:17:31 作者：戀喵大鯉魚

Go語言并沒有提供用于刪除元素的語法或接口,而是通過利用切片本身的特性來刪除元素—追加元素,這篇文章主要給大家介紹了關(guān)于Golang切片刪除指定元素的三種方法,需要的朋友可以參考下

前言

Go 并沒有提供刪除切片元素專用的語法或函數(shù)，需要使用切片本身的特性來刪除元素。

刪除切片指定元素一般有如下幾種方法，本文以 []int 為例給出具體實現(xiàn)。

1.截取法（修改原切片）

這里利用對 slice 的截取刪除指定元素。注意刪除時，后面的元素會前移，所以下標 i 應(yīng)該左移一位。

// DeleteSlice1 刪除指定元素。
func DeleteSlice1(a []int, elem int) []int {
	for i := 0; i < len(a); i++ {
		if a[i] == elem {
			a = append(a[:i], a[i+1:]...)
			i--
		}
	}
	return a
}

2.拷貝法（不改原切片）

這種方法最容易理解，重新使用一個 slice，將要刪除的元素過濾掉。缺點是需要開辟另一個 slice 的空間，優(yōu)點是容易理解，而且不會修改原 slice。

// DeleteSlice2 刪除指定元素。
func DeleteSlice2(a []int, elem int) []int {
	tmp := make([]int, 0, len(a))
	for _, v := range a {
		if v != elem {
			tmp = append(tmp, v)
		}
	}
	return tmp
}

3.移位法（修改原切片）

3.1 方式一

利用一個下標 index，記錄下一個有效元素應(yīng)該在的位置。遍歷所有元素，當(dāng)遇到有效元素，將其移動到 index 且 index 加一。最終 index 的位置就是所有有效元素的下一個位置，最后做一個截取就行了。這種方法會修改原來的 slice。

該方法可以看成對第一種方法截取法的改進，因為每次指需移動一個元素，性能更加。

// DeleteSlice3 刪除指定元素。
func DeleteSlice3(a []int, elem int) []int {
	j := 0
	for _, v := range a {
		if v != elem {
			a[j] = v
			j++
		}
	}
	return a[:j]
}

3.2 方式二

創(chuàng)建了一個 slice，但是共用原始 slice 的底層數(shù)組。這樣也不需要額外分配內(nèi)存空間，直接在原 slice 上進行修改。

// DeleteSlice4 刪除指定元素。
func DeleteSlice4(a []int, elem int) []int {
	tgt := a[:0]
	for _, v := range a {
		if v != elem {
			tgt = append(tgt, v)
		}
	}
	return tgt
}

4.性能對比

假設(shè)我們的切片有 0 和 1，我們要刪除所有的 0。

這里分別對長度為 10、100、1000 的切片進行測試，來上下上面四種實現(xiàn)的性能差異。

生成切片函數(shù)如下：

func getSlice(n int) []int {
	a := make([]int, 0, n)
	for i := 0; i < n; i++ {
		if i%2 == 0 {
			a = append(a, 0)
			continue
		}
		a = append(a, 1)
	}
	return a
}

基準測試代碼如下：

func BenchmarkDeleteSlice1(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_ = DeleteSlice1(getSlice(10), 0)
	}
}
func BenchmarkDeleteSlice2(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_ = DeleteSlice2(getSlice(10), 0)
	}
}
func BenchmarkDeleteSlice3(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_ = DeleteSlice3(getSlice(10), 0)
	}
}
func BenchmarkDeleteSlice4(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_ = DeleteSlice4(getSlice(10), 0)
	}
}

測試結(jié)果如下：

原切片長度為 10：

go test -bench=. main/slice
goos: windows
goarch: amd64
pkg: main/slice
cpu: Intel(R) Core(TM) i7-9700 CPU @ 3.00GHz
BenchmarkDeleteSlice1-8 17466486 65.07 ns/op
BenchmarkDeleteSlice2-8 14897282 85.22 ns/op
BenchmarkDeleteSlice3-8 21952129 50.78 ns/op
BenchmarkDeleteSlice4-8 22176390 54.68 ns/op
PASS
ok main/slice 5.427s

原切片長度為 100：

BenchmarkDeleteSlice1-8 1652146 762.1 ns/op
BenchmarkDeleteSlice2-8 2124237 578.4 ns/op
BenchmarkDeleteSlice3-8 3161318 359.9 ns/op
BenchmarkDeleteSlice4-8 2714158 423.7 ns/op

原切片長度為 1000：

BenchmarkDeleteSlice1-8 56067 21915 ns/op
BenchmarkDeleteSlice2-8 258662 5007 ns/op
BenchmarkDeleteSlice3-8 432049 2724 ns/op
BenchmarkDeleteSlice4-8 325194 3615 ns/op