当前位置:   article > 正文

golang删除slice中特定条件的元素,优化版_从slice中移除元素

从slice中移除元素

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

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

第一种方法:

  1. func DeleteSlice(a []int) []int{
  2. for i := 0; i < len(a); i++ {
  3. if a[i] == 0 {
  4. a = append(a[:i], a[i+1:]...)
  5. i--
  6. }
  7. }
  8. return a
  9. }

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

第二种方法:

  1. func DeleteSlice1(a []int) []int {
  2. ret := make([]int, 0, len(a))
  3. for _, val := range a {
  4. if val == 1 {
  5. ret = append(ret, val)
  6. }
  7. }
  8. return ret
  9. }

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

第三种方法:

  1. func DeleteSlice2(a []int) []int{
  2. j := 0
  3. for _, val := range a {
  4. if val == 1 {
  5. a[j] = val
  6. j++
  7. }
  8. }
  9. return a[:j]
  10. }

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

 

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

  1. package main
  2. import (
  3. "testing"
  4. )
  5. func handle(data []int) {
  6. return
  7. }
  8. const N = 100
  9. func getSlice()[]int {
  10. a := []int{}
  11. for i := 0; i < N; i++ {
  12. if i % 2 == 0 {
  13. a = append(a, 0)
  14. } else {
  15. a = append(a, 1)
  16. }
  17. }
  18. return a
  19. }
  20. func BenchmarkDeleteSlice(b *testing.B) {
  21. for i := 0; i < b.N; i++ {
  22. data := DeleteSlice(getSlice())
  23. handle(data)
  24. }
  25. }
  26. func BenchmarkDeleteSlice1(b *testing.B) {
  27. for i := 0; i < b.N; i++ {
  28. data := DeleteSlice1(getSlice())
  29. handle(data)
  30. }
  31. }
  32. func BenchmarkDeleteSlice2(b *testing.B) {
  33. for i := 0; i < b.N; i++ {
  34. data := DeleteSlice2(getSlice())
  35. handle(data)
  36. }
  37. }

测试结果如下(slice大小为100):

加大slice大小进行测试(slice大小为10000):

继续加大(slice大小为100000)

slice大小为10^6:

可以看出:

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

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

本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/空白诗007/article/detail/850397
推荐阅读
相关标签
  

闽ICP备14008679号