欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

golang删除slice中某些符合条件的元素

程序员文章站 2022-07-12 14:19:37
...

假设我们的切片有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进行修改。

性能测试

第一种方法在slice大小比较小时,比第2、3种方法慢一倍左右。但是slice大小变大时,性能显著下降。

第2种方法和第3种方法差距基本处于同一量级,但是第3种方法稍快一些。但是当slice大小增加到10^6级别时,第三种方法的优势就显现出来。

相关标签: 后端