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