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

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