发现问题
大家在码代码的过程中,经常会遇到在循环中移除指定元素的需求。按照常规的思路,直接一个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为数组下标,个人觉得这是最简洁、高效的代码实现了。