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