STL容器中erase方法的不同陷阱

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

先以map为例:

[cpp]   view plain copy

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

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

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

[c-sharp]   view plain copy

1. for(TStrMapIter iter= strmap.begin(); iter!= strmap.end();++iter)  
2. {  
3.    if ("somevalue" == iter->second )  
4.    {  
5.      strmap.erase(iter);  
6.    }  
7. }  

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

 

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

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

 

[cpp]   view plain copy

1. for(TStrMapIter iter= strmap.begin(); iter!= strmap.end();)  
2. {  
3.    if ("somevalue" == iter->second )  
4.    {  
5.      strmap.erase(iter++);  
6.    }  
7.    else  
8.    {  
9.      ++iter;   
10.    }  
11. }  

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

[cpp]   view plain copy

1. for(TStrMapIter iter= strmap.begin(); iter!= strmap.end();)  
2. {  
3.    if ("somevalue" == iter->second )  
4.    {  
5.      iter = strmap.erase(iter);  
6.    }  
7.    else  
8.    {  
9.      ++iter;  
10.    }  
11. }  

 

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

[cpp]   view plain copy

1. void test_vector_erase()  
2. {  
3.     typedef std::vector<int> TIntVec;  
4.     typedef TIntVec::iterator TIntVecIter;  
5.     TIntVec vec;  
6.   
7.     vec.push_back(1);  
8.     vec.push_back(2);  
9.     vec.push_back(3);  
10.     vec.push_back(4);  
11.   
12.     for (TIntVecIter iter = vec.begin(); iter != vec.end();)  
13.     {  
14.         std::cout << *iter << std::endl;  
15.         if (0 == *iter % 2)  
16.         {  
17.             vec.erase(iter++);  
18.         }  
19.         else  
20.         {  
21.             ++iter;  
22.         }  
23.     }  
24. }  

  

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

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