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

问题

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

    (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);
    })();

运行结果如下：

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

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

解决

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

姿势一

    (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);
    })();

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

姿势二

    (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);
    })();

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

姿势三

    (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为数组下标，个人觉得这是最简洁、高效的代码实现了。