蘭陵N梓記

一指流沙,程序年华


  • 首页

  • 归档

  • 关于

  • 搜索
close

c++ STL容器erase方法的不同陷阱

时间: 2009-07-01   |   分类: 技术     |   阅读: 790 字 ~2分钟

陷阱

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这样的链式存储结构。我们一般可以有两种解决办法:

方法一

使用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;   
   }  
}

方法二

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方法,那是一定要小心再小心,我也是被它戏弄了一下之后,才明白其中容易被忽视的这些细节。

#c++#
面向对象的设计原则
c++的仿函数与动态语言的闭包
微信扫一扫交流

标题:c++ STL容器erase方法的不同陷阱
作者:兰陵子
关注:lanlingthink(览聆时刻)
声明:自由转载-非商用-非衍生-保持署名(创作共享3.0许可证)

  • 文章目录
  • 站点概览
兰陵子

兰陵子

Programmer & Architect

164 日志
4 分类
57 标签
GitHub 知乎
    • 陷阱
    • 解决办法
      • 方法一
      • 方法二
© 2009 - 2022 蘭陵N梓記
Powered by - Hugo v0.101.0
Theme by - NexT
0%