在List的remove操作中常有同学会这么写
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("ccc");
list.add("aaa");
list.add("bbb");
list.add("ddd");
list.add("fff");
for(int i=0;i<list.size();i++){
String each = list.get(i);
if("bbb".equals(each)){
list.remove(i);
}
}
System.out.println(list.size());
}
嗯,很不错~代码看上去没有任何问题,运行测试也很正确,但是如果等于bbb或ddd这两个都要remove呢?那么自然会这么写~
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("ccc");
list.add("aaa");
list.add("bbb");
list.add("ddd");
list.add("fff");
for(int i=0;i<list.size();i++){
String each = list.get(i);
if("bbb".equals(each)||"ddd".equals(each)){
list.remove(i);
}
}
System.out.println(list.size());
}
然而会发现只移除了bbb。。。那么如何移除正确呢?最简单的写成i--即可
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("ccc");
list.add("aaa");
list.add("bbb");
list.add("ddd");
list.add("fff");
for(int i=0;i<list.size();i++){
String each = list.get(i);
if("bbb".equals(each)||"ddd".equals(each)){
list.remove(i--);//i--
}
}
System.out.println(list.size());
}
或者用Iterator也可以
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("ccc");
list.add("aaa");
list.add("bbb");
list.add("ddd");
list.add("fff");
Iterator<String> it = list.iterator();
while(it.hasNext()){
String each = it.next();
if("bbb".equals(each)||"ddd".equals(each)){
it.remove();
}
}
System.out.println(list.size());
}