Golang slice 删除元素性能比较

写了两种对一个 slice 中删除特定元素的方法，并做了性能对比，在这里记录一下。

假设我们的切片有 0 和 1，我们要删除所有的 0，此处有三种方法：

第一种方法：

func DeleteSlice(a []int) []int{
	for i := 0; i < len(a); i++ {
		if a[i] == 0 {
			a = append(a[:i], a[i+1:]...)
			i--
		}
	}
	return a
}

解释：这里利用常见的方法对 slice 中的元素进行删除，注意删除时，后面的元素前移，i 应该后移一位。

第二种方法：

func DeleteSlice1(a []int) []int {
	ret := make([]int, 0, len(a))
	for _, val := range a {
		if val == 1 {
			ret = append(ret, val)
		}
	}
	return ret
}

解释：这种方法最容易理解，重新使用一个 slice，将不合理的过滤掉。缺点是需要开辟另一个 slice 的空间，优点是容易理解，而且不对原来的 slice 进行操作。

第三种方法：

func DeleteSlice2(a []int) []int{
	j := 0
	for _, val := range a {
		if val == 1 {
			a[j] = val
			j++
		}
	}
	return a[:j]
}

解释：这里利用一个 index，记录应该下一个有效元素应该在的位置，遍历所有元素，当遇到有效元素，index 加一，否则不加，最终 index 的位置就是所有有效元素的下一个位置。最后做一个截取就行了。这种方法会对原来的 slice 进行修改。

这里对三种方法做了性能测试，测试代码如下：

package main
 
import (
	"testing"
)
 
func handle(data []int) {
	return
}
const N = 100
 
func getSlice()[]int {
	a := []int{}
	for i := 0; i < N; i++ {
		if i % 2 == 0 {
			a = append(a, 0)
		} else {
			a = append(a, 1)
		}
	}
	return a
}
 
func BenchmarkDeleteSlice(b *testing.B) {
	for i := 0; i < b.N; i++ {
		 data := DeleteSlice(getSlice())
		 handle(data)
	}
}
 
func BenchmarkDeleteSlice1(b *testing.B) {
	for i := 0; i < b.N; i++ {
		data := DeleteSlice1(getSlice())
		handle(data)
	}
}
 
func BenchmarkDeleteSlice2(b *testing.B) {
	for i := 0; i < b.N; i++ {
		data := DeleteSlice2(getSlice())
		handle(data)
	}
}