Programmer

循环删除数组元素的几种姿势

问题

  在码代码的过程中，经常会遇到在循环中移除指定元素的需求。按照常规的思路，直接一个for循环，然后在循环里面来个if判断，在判断中删除掉指定元素即可。但是实际情况往往不会像预想的那样顺利运行。下面以一段Javascript代码为例演示这一过程。

1
2
3
4
5
6
7
8
9
10
11
12
13
(function () {
    var arr = [1,2,2,3,4,5];
    var len = arr.length;
    for(var i=0;i<len;i++){
        //打印数组中的情况，便于跟踪数组中数据的变化
        console.log(i+"="+arr[i]);
        //删除掉所有为2的元素
        if(arr[i]==2){
            arr.splice(i,1);
        }
    }
    console.log(arr);
})();

运行结果如下：

  从最终的结果可以看到实际上只删除掉了匹配的其中一个元素，而另外一个元素还存在。

  从打印出的运行过程不难发现，原因为当删除掉了一个元素后，数组的索引发生的变化，造成了程序的异常。

解决

找到了问题的原因，就不难解决问题了。

姿势一

1
2
3
4
5
6
7
8
9
10
11
12
13
14
(function () {
    var arr = [1,2,2,3,4,5];
    var len = arr.length;
    for(var i=0;i<len;i++){
        //打印数组中的情况，便于跟踪数组中数据的变化
        console.log(i+"="+arr[i]);
        //删除掉所有为2的元素
        if(arr[i]==2){
            //注意对比这行代码：删除元素后调整i的值
            arr.splice(i--,1);
        }
    }
    console.log(arr);
})();

上面的代码看起来不大好理解，有没有看起来更易于理解的代码呢？请看下面

姿势二

1
2
3
4
5
6
7
8
9
10
11
12
(function () {
    var arr = [1,2,2,3,4,5];
    var len = arr.length-1;
    //start from the top
    for(var i=len;i>=0;i--){
        console.log(i+"="+arr[i]);
        if(arr[i]==2){
            arr.splice(i,1);
        }
    }
    console.log(arr);
})();

从后往前遍历可以有效解决问题，也容易理解，那么还有没有跟简洁的实现呢？接着看下面代码

姿势三

1
2
3
4
5
6
7
8
9
10
11
(function () {
    var arr = [1,2,2,3,4,5];
    var i = arr.length;
    while(i--){
        console.log(i+"="+arr[i]);
        if(arr[i]==2){
            arr.splice(i,1);
        }
    }
    console.log(arr);
})();

使用while(i--),i为数组下标，个人觉得这是最简洁、高效的代码实现了。

作者公众号：

留言

欢迎交流想法。留言会通过 GitHub Issues 保存，首次使用需要登录 GitHub。