热门标签 | HotTags
当前位置:  开发笔记 > 运维 > 正文

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

set和map是由红黑树来实现的,当erase的时候迭代器就失效了,也就是说我们要在迭代器失效之前保留一个副本,根据这个副本我们才能继续遍历下一个元素

代码如下:

#include
#include
#include
#include
using namespace std;

void Remove1(vector &vec, int num)
{
 vector::iterator iter;
 for (iter=vec.begin(); iter!=vec.end(); ++iter)
 {
  if (*iter == num)
  {
   vec.erase(iter);
  }
 }
}

void Remove2(list &lst, int num)
{
 list::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 vec(arr, arr+num);
 list lst(arr, arr+num);

 Remove1(vec, 5);
 copy(vec.begin(), vec.end(), ostream_iterator(cout, " "));

 //Remove2(lst, 5);
 //copy(lst.begin(), lst.end(), ostream_iterator(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 &vec, int num)
{
 std::vector::iterator iter = vec.begin();
 while (iter != vec.end())
 {
  if (*iter == num)
  {
   iter = vec.erase(iter);
  }
  else
  {
   ++iter;
  }
 }
}

// 删除某一类元素, 比如:偶数
void Remove3(std::set& sets)
{
 std::set::iterator iter = sets.begin();
 while (iter != sets.end())
 {
  if (0 == (*iter)%2)
  {
   // 注意这里不能写成++iter,后面说明原因
   sets.erase(iter++);
  }
  else
  {
   ++iter;
  }
 }
}

void Remove4(std::map& maps)
{
 std::map::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++


推荐阅读
author-avatar
yyyyyyyyyyyyy燁
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有