[C++] STL容器中erase方法的不同陷阱

STL中的容器都有erase方法，容器的存储分为顺序存储(如vector)与链式存储(如list,map)。

先以map为例:

typedef std::map<std::string, std::string> TStrMap; typedef TStrMap::iterator TStrMapIter; TStrMap strmap; TStrMapIter iter = strmap.find("somekey"); strmap.erase(iter);

这样使用erase方法没有任何问题，删除一个单结节之后，stl中的iterator都是与其中的数据元素关联的，关联的元素删除之后，iter已就失效，iter理解为指向元素的指针，那删除之后可以简单理解为已是一个野指针。

但有时我们一不注意，却会这样使用，这是错误的。

for(TStrMapIter iter= strmap.begin(); iter!= strmap.end();++iter) { if ("somevalue" == iter->second ) { strmap.erase(iter); } }

iter所指的元素删除之后，++iter是错误的，会导致程序的未知结果，iter一般是不会移到指向下一个元素。

 

对于map与list这样的链式存储结构。我们一般可以有两种解决办法

1. 使用erase(iter++)，因为iter2 = iter++是iter先移到指向下一个节点，而iter2还是指向当前的节点。注意理解iter++与++iter的区别。

 for(TStrMapIter iter= strmap.begin(); iter!= strmap.end();) { if ("somevalue" == iter->second ) { strmap.erase(iter++); } else { ++iter; } }

2. erase的返回值会指向下一个节点

for(TStrMapIter iter= strmap.begin(); iter!= strmap.end();) { if ("somevalue" == iter->second ) { iter = strmap.erase(iter); } else { ++iter; } }

 

但对于顺序存储的vector也可以使用上述两种方法吗？很遗憾，第一种用法却是错误的，但第二种用法是正确的。因为顺序存储的容器一旦erase时，会涉及到数据移动，iterator所指的位置还是那个位置，但元素却移动了，iter++之后已不再你想要的元素位置了。

void test_vector_erase() { typedef std::vector<int> TIntVec; typedef TIntVec::iterator TIntVecIter; TIntVec vec; vec.push_back(1); vec.push_back(2); vec.push_back(3); vec.push_back(4); for (TIntVecIter iter = vec.begin(); iter != vec.end();) { std::cout << *iter << std::endl; if (0 == *iter % 2) { vec.erase(iter++); } else { ++iter; } } }  

它输出的结果却是未知的，我的测试环境为1，2，4，4。你可能发现原因，当删除2元素时，3往前移了，而iter++不是指到3，还是指到4了。

当你使用STL容器中erase方法，那是一定要小心再小心，我也是被它戏弄了一下之后，才明白其中容易被忽视的这些细节。