vector中erase用法注意事项
http://blog.csdn.net/dgyanyong/article/details/21268469
vector::erase():从指定容器删除指定位置的元素或某段范围内的元素
vector::erase()方法有两种重载形式
如下:
iterator erase( iterator _Where);
iterator erase( iterator _First, iterator _Last);
如果是删除指定位置的元素时:
返回值是一个迭代器,指向删除元素下一个元素;
如果是删除某范围内的元素时:返回值也表示一个迭代器,指向最后一个删除元素的下一个元素;
看下面的程序,目的是删除数组里面的所有值为6的元素:
- #include<iostream>
- #include<vector>
- using namespace std;
- int main()
- {
- vector<int> array;
- array.push_back(1);
- array.push_back(6);
- array.push_back(3);
- array.push_back(6);
- array.push_back(6);
- array.push_back(2);
- vector<int>::iterator itor;
- vector<int>::iterator itor2;
- for(itor=array.begin();itor!=array.end();)
- {
- if(6==*itor)
- {
- itor2=itor;
- array.erase(itor2);
- }
- itor++;
- }
- itor=array.begin();
- for(itor=array.begin();itor!=array.end();)
- {
- cout<<(*itor++);
- }
- getchar();
- return 0;
- }
运行结果输出1362,可见其中一个6并未删除,这是迭代器的问题。
原因在于erase以后,itor已经指向下一个元素了,不应该在itor++,否则会跳过下一个元素,即连续两个6时跳过了第二个6.
另外,在itor2=itor时,两个itor是一样的,这样做并无意义。可修改如下:
- vector<int>::iterator itor;
- // vector<int>::iterator itor2;
- for(itor=array.begin();itor!=array.end();)
- {
- if(6==*itor)
- {
- // itor2=itor;
- array.erase(itor);
- }
- else
- {
- itor++;
- }
- }
或者:
- vector<int>::iterator itor;
- for(itor=array.begin();itor!=array.end();itor++)
- {
- if(6==*itor)
- {
- array.erase(itor);
- itor--;
- }
- }
也可以使用remove方法:
array.earse( remove(array.begin(), array.end(),6), array.end() );