大家注意vector, list, set, map成员函数erase
#include <iostream>
#include <vector>
#include <list>
#include <iterator>
using namespace std;
void Remove1(vector<int> &vec, int num)
{
vector<int>::iterator iter;
for (iter=vec.begin(); iter!=vec.end(); ++iter)
{
if (*iter == num)
{
vec.erase(iter);
}
}
}
void Remove2(list<int> &lst, int num)
{
list<int>::iterator iter;
for (iter=lst.begin(); iter!=lst.end(); ++iter)
{
if (*iter == num)
{
lst.erase(iter);
}
}
}
int main(void)
{
int arr[] = {1, 3, 5, 5, 7, 9};
int num = sizeof(arr) / sizeof(arr[0]);
vector<int> vec(arr, arr+num);
list<int> lst(arr, arr+num);
Remove1(vec, 5);
copy(vec.begin(), vec.end(), ostream_iterator<int>(cout, " "));
//Remove2(lst, 5);
//copy(lst.begin(), lst.end(), ostream_iterator<int>(cout, " "));
return 0;
}
请看上面的Remove1、Remove2这两个函数是删除容器中所有为num的元素，乍一看好像没什么问题，其实是错的。
它们编译都没什么问题，但是运行有问题：
先注释Remove2这两行， 编译运行， 得到结果是：1 3 5 7 9
显然结果是错误的，因为还有个5没删掉。造成这样结果的原因是vec.erase之后iter会自加，执行下次for循环时++iter，这样就跳过了与所删除元素相邻的元素了。
再来看下Remove2， 注释掉Remove1， 删除Remove2的注释。
运行出错，vs2005 Expression:list iterator not incrementable，说明list迭代器++iter有问题。
list是一个双向链表，在erase(iter)之后iter失效了，printf("0x%x", iter);得到结果为0，只有erase的返回值才能指向下一个元素。
Remove1的正确用法是：vec.erase(remove(vec.begin(), vec.end(), 5), vec.end());
Remove2的正确用法是：lst.remove(5);
下面是另一种正确写法：
void Remove1(std::vector<int> &vec, int num)
{
std::vector<int>::iterator iter = vec.begin();
while (iter != vec.end())
{
if (*iter == num)
{
iter = vec.erase(iter);
}
else
{
++iter;
}
}
}
// 删除某一类元素， 比如：偶数
void Remove3(std::set<int>& sets)
{
std::set<int>::iterator iter = sets.begin();
while (iter != sets.end())
{
if (0 == (*iter)%2)
{
// 注意这里不能写成++iter，后面说明原因
sets.erase(iter++);
}
else
{
++iter;
}
}
}
void Remove4(std::map<int, int>& maps)
{
std::map<int, int>::iterator iter = maps.begin();
while (iter != maps.end())
{
if (0 == (iter->first)%2)
{
maps.erase(iter++);
}
else
{
++iter;
}
}
}
set和map是由红黑树来实现的，当erase的时候迭代器就失效了，也就是说我们要在迭代器失效之前保留一个副本，根据这个副本我们才能继续遍历下一个元素。i++和++i很明显前者符合我们的要求所以在erase里面是iter++
上一篇：C++多态的实现机制深入理解
栏 目：C代码
下一篇：C++ STL入门教程(2) list双向链表使用方法（附程序代码）
本文标题：大家注意vector, list, set, map成员函数erase
本文地址：http://www.codeinn.net/misctech/101642.html
