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

来源:互联网 发布:金山软件 港股 编辑:程序博客网 时间:2024/05/15 20:18

我们通过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下这样写代码是无法通过编译的

0 0
原创粉丝点击