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

From: http://blog.csdn.net/tujiaw/article/details/7770498


  1. #include <iostream>  
  2. #include <vector>  
  3. #include <list>  
  4. #include <iterator>  
  5. using namespace std;  
  6.   
  7. void Remove1(vector<int> &vec, int num)  
  8. {  
  9.     vector<int>::iterator iter;  
  10.     for (iter=vec.begin(); iter!=vec.end(); ++iter)  
  11.     {  
  12.         if (*iter == num)  
  13.         {  
  14.             vec.erase(iter);  
  15.         }  
  16.     }  
  17. }  
  18.   
  19. void Remove2(list<int> &lst, int num)  
  20. {  
  21.     list<int>::iterator iter;  
  22.     for (iter=lst.begin(); iter!=lst.end(); ++iter)  
  23.     {  
  24.         if (*iter == num)  
  25.         {  
  26.             lst.erase(iter);  
  27.         }  
  28.     }  
  29. }  
  30.   
  31. int main(void)  
  32. {  
  33.     int arr[] = {1, 3, 5, 5, 7, 9};  
  34.     int num = sizeof(arr) / sizeof(arr[0]);  
  35.     vector<int> vec(arr, arr+num);  
  36.     list<int> lst(arr, arr+num);  
  37.   
  38.     Remove1(vec, 5);  
  39.     copy(vec.begin(), vec.end(), ostream_iterator<int>(cout, " "));  
  40.       
  41.     //Remove2(lst, 5);  
  42.     //copy(lst.begin(), lst.end(), ostream_iterator<int>(cout, " "));  
  43.   
  44.     return 0;  
  45. }  

请看上面的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);

下面是另一种正确写法:

  1. void Remove1(std::vector<int> &vec, int num)  
  2. {  
  3.     std::vector<int>::iterator iter = vec.begin();  
  4.     while (iter != vec.end())  
  5.     {  
  6.         if (*iter == num)  
  7.         {  
  8.             iter = vec.erase(iter);  
  9.         }  
  10.         else  
  11.         {  
  12.             ++iter;  
  13.         }  
  14.     }  
  15. }  
  16.   
  17. // 删除某一类元素, 比如:偶数  
  18. void Remove3(std::set<int>& sets)  
  19. {  
  20.     std::set<int>::iterator iter = sets.begin();  
  21.     while (iter != sets.end())  
  22.     {  
  23.         if (0 == (*iter)%2)  
  24.         {  
  25.             // 注意这里不能写成++iter,后面说明原因  
  26.             sets.erase(iter++);  
  27.         }  
  28.         else  
  29.         {  
  30.             ++iter;  
  31.         }  
  32.     }  
  33. }  
  34.   
  35. void Remove4(std::map<intint>& maps)  
  36. {  
  37.     std::map<intint>::iterator iter = maps.begin();  
  38.     while (iter != maps.end())  
  39.     {  
  40.         if (0 == (iter->first)%2)  
  41.         {  
  42.             maps.erase(iter++);  
  43.         }  
  44.         else  
  45.         {  
  46.             ++iter;  
  47.         }  
  48.     }  
  49. }  

set和map是由红黑树来实现的,当erase的时候迭代器就失效了,也就是说我们要在迭代器失效之前
保留一个副本,根据这个副本我们才能继续遍历下一个元素。i++和++i很明显前者符合我们的要求所以
在erase里面是iter++
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值