您的位置:首页 > 编程语言 > C语言/C++

C++ STL遍历map的时候如何删除其中的element

2015-08-08 16:14 393 查看
我们通过map的erase(iterator it)方法删除元素的时候,如果此时erase处于遍历map的代码中,那么调用erase就需要小心一些。因为erase会导致输入参数iterator变的无效,从而影响后续的it++遍历map的逻辑。

简单做法是,先将要删除的it保存下来,然后将用于遍历map的it指向下一个位置,然后删除掉保存下来的it。如下面代码所示:

#include <map>
#include <iostream>
using namespace std;

int main()
{
map<int, int> map1;
map<int, int>::iterator mapit;
map<int, int>::iterator saveit;

map1[1] = 2;
map1[2] = 3;
map1[3] = 4;
map1[4] = 5;

mapit = map1.begin();
while (mapit != map1.end()) {
cout << "Element key: " << mapit->first << ", value: " << mapit->second << endl;
if (mapit->first == 2) {
saveit = mapit;
mapit++;
map1.erase(saveit);
continue;
}
mapit++;
}

cout << "Map size: " << map1.size() << endl;
return 0;
}


需要注意的是,这里windows的STL(windows C++编译器带的STL)和linux上的STL(gcc的STL)实现不同。

windows的STL中,map的erase方法会返回一个iterator,这个iterator指向的是当前被删除的iterator后面的iterator,所以这样的话,只需要将用于循环的iterator赋成erase函数的返回值就可以了。参考上面代码,就是这样:

mapit = map1.erase(mapit);然后continue就可以。

但是Linux下这样写代码是无法通过编译的
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: