说明
之前做的go项目，遇到一个关于数组的增删效率问题：
循环数组--判断--删除数组--得到需要的数组
示例：
如果我需要在arr1中过滤掉arr2
第一次写的方法如下
arr1:=[]int{1,2,3......,999,1000}
arr2:=[]int{5,10,15......,995,1000}
for k, v := range arr1{
for _, vv := range arr2{
if v == vv {
arr1= append(arr1[:k], arr1[k+1:]...)
}
}
}
这样写会出现错误，因为删除一个数之后，数标就对不上了，因此需要倒序删除：
for i := len(arr1) - 1; i >= 0; i-- {
v := arr1[i]
for _, vv := range arr2 {
if v == vv {
arr1 = append(arr1[:i], arr1[i+1:]...)
}
}
}
还有一种方法，就是不删除，用一个新的temparr来append：
arr1 := []int{1, 2, 3, 5, 6, 7, 8, 9}
var temp []int
for _, v := range arr1 {
exist := false
for _, vv := range arr2 {
if v == vv {
exist = true
}
}
if !exist {
temp = append(temp, v)
}
}
对比
测试源码 github-demo
通过golang的Benchmark来进行测试，
结果如下：
第一种，删除方法
第二种，temp添加方法
结果
这个测试用例在1s中内运行了20000次，每次调用大约用了86660ns、97710ns
可以看到 对数组使用删除的方式会增加效率，当然，我觉得没什么太大的区别。。仅仅用来测试一下
更新：
如果 数组的长度越大，运行时间会越长，如果将arr2先转换为map，则时间复杂度由n^2 变为 2n，少了一个数量级。
方法汇总如下
package main
import (
"fmt"
"testing"
)
var (
arr2 []int
arrMap map[int]bool
)
func init() {
arrMap = make(map[int]bool, 0)
for i := 1000; i < 10000; i++ {
arr2 = append(arr2, i)
arrMap[i] = true
}
}
func main() {
One()
Two()
Three()
Four()
}
// panic
func One() {
arr1 := []int{1, 2, 3, 5, 6, 7, 8, 9}
defer func() {
if err := recover(); err != nil {
fmt.Println("panic ", err)
}
}()
for k, v := range arr1 {
for _, vv := range arr2 {
if v == vv {
arr1 = append(arr1[:k], arr1[k+1:]...)
}
}
}
fmt.Println(arr1, len(arr1))
}
func Two() []int {
arr1 := []int{1, 2, 3, 5, 6, 7, 8, 9}
for i := len(arr1) - 1; i >= 0; i-- {
v := arr1[i]
for _, vv := range arr2 {
if v == vv {
arr1 = append(arr1[:i], arr1[i+1:]...)
}
}
}
return arr1
}
func Three() []int {
arr1 := []int{1, 2, 3, 5, 6, 7, 8, 9}
var temp []int
for _, v := range arr1 {
exist := false
for _, vv := range arr2 {
if v == vv {
exist = true
}
}
if !exist {
temp = append(temp, v)
}
}
return arr1
}
func Four() []int {
arr1 := []int{1, 2, 3, 5, 6, 7, 8, 9}
var temp []int
for _, v := range arr1 {
if !arrMap[v] {
temp = append(temp, v)
}
}
return temp
}
func BenchmarkTwo(b *testing.B) {
for n := 0; n < b.N; n++ {
Two()
}
}
func BenchmarkThree(b *testing.B) {
for n := 0; n < b.N; n++ {
Three()
}
}
func BenchmarkFour(b *testing.B) {
for n := 0; n < b.N; n++ {
Four()
}
}
go test -bench=.
➜ array go test -bench=.
goos: linux
goarch: amd64
pkg: test/array
BenchmarkTwo-12 50000 33427 ns/op
BenchmarkThree-12 50000 33378 ns/op
BenchmarkFour-12 10000000 183 ns/op
PASS
ok test/array 6.041s
