集合使用迭代器遍历删除元素的问题
记一次面试问题:
在集合中删除元素再普通不过,一般想到的就是for循环遍历,在里面删除指定的元素,如下所示:
public class Main {
public static void main(String[] args) {
ArrayList<Integer> lists = new ArrayList<>();
for (int i = 0; i < 5; i++) {
lists.add(i);
}
printList(lists);
for (int i = 0; i < lists.size(); i++) {
if(2 == lists.get(i)){
lists.remove(2);
}
}
System.out.println();
printList(lists);
}
private static void printList(List<Integer> lists){
for (int i = 0; i <lists.size(); i++) {
if(i == 0){
System.out.print("{ " +lists.get(i) +", ");
}else if(i == lists.size() - 1){
System.out.print(lists.get(i) +" }");
}else{
System.out.print(lists.get(i) +", ");
}
}
}
}
如果使用集合的迭代器进行遍历删除指定元素会出现什么问题呢? 首先在迭代器中删除指定元素使用list的remove方法时,如下:
可以看到程序报错了,这种操作是不允许的。如何改善呢,在迭代器中遍历集合想要删除元素时不能使用集合的remove方法进行删除,而应该使用迭代器本身的remove方法进行操作,这样就不会报错了。