注意vector, list, set, map成员函数erase

来源:互联网 发布:xilinx linux系统类型 编辑:程序博客网 时间:2024/05/16 19:43
#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++

原创粉丝点击