my_container.erase(iter);
其中my_container是STL的某种容器,iter是指向这个容器中某个元素的迭代器。如果不是在for,while循环中,这种方式删除元素没有问题,如果是在for,while中对m_container迭代,删除其中符合条件的所有元素,就可能出现问题。如果是在for,while中对m_container迭代,删除其中符合条件的所有元素,就可能出现问题。
问题是:
在迭代容器的时候删除元素,可能导致迭代器失效(invalidation of iterators),产生未定义行为(undefined behavior);例如,对某个迭代器解引用所获得的值并不是执行erase()前这个迭代器指向的值,还有可能对未指向任何元素的迭代器的解引用赋值而引发程序crash。
类似的问题代码像这样:
01.std::vector<int> my_container; 02.for (int i = 0; i < 100; ++i) { 03. my_container.push_back(i); 04.bsp; } 05. 06.std::vector<int>::iterator it = my_container.begin(); 07.for (it != my_container.end(); it++) { 08. if (*it % 2 == 1) { 09. my_container.erase(it); 10. } 11.}
my_container.erase(it)之后,it及其后面的迭代器已经失效,不应该再使用这些迭代器。再执行it++,其行为是未定义的。
其他容器也会遇到迭代器失效的问题:
对于vector 被删除元素的迭代器以及指向后面元素的迭代器全部失效。
对于deque 在首部或尾部删除元素则只会使指向被删除元素的迭代器失效,任何其他位置的插入和删除操作将使指向该容器元素的 所有迭代器失效。
对于list 仅有指向被删除元素的迭代器失效。
对于(mulit)map ,(multi)set 仅有指向被删除元素的迭代器失效。
所以Golden Rule是:尽量不要使用容器的插入删除操作之前的迭代器。
为什么不同容器迭代器失效情况有差别?这与实现各容器的数据结构有关。
如何在迭代容器时删除其中的元素?各容器通用的做法如下:
std::vector<int>::iterator it = my_container.begin(); 02. for (it != my_container.end();/**blank*/ ) { 03. if (*it % 2 == 1) { 04. my_container.erase(it++); 05. } 06. else{ 07. it++; 08. } 09. }
my_container.erase(it++) 巧妙得在执行erase()之前,it 先自增,指向被删除元素后面的元素,而给erase()传递的是未自增的it迭代器,
以定位要删除的元素。如果元素的值为奇数,则删除此元素,it指向下一个元素,如果元素的值为偶数,则检查下一个元素的值。整个迭代过程中
迭代器就不会失效了。
上段代码中两个不同分支出现了i++操作,下面的代码示例显示了如何防止遗忘其中任何一个分支的i++操作。
MyContainer::iterator it = myContainer.begin(); 02. While(it != myContainer.end()){ 03. MyContainer::iterator curIt = it; 04. if (*curIt == matchingValue) { 05. myContainer.erase(curIt); 06. } 07. }
对于vector ,deque, list, 另一种可行的方式是:
01.std::vector<int>::iterator it = my_container.begin(); 02.for (it != my_container.end();/**blank*/ ) { 03. if (*it % 2 == 1) { 04. it = my_container.erase(it); 05. } 06. else{ 07. it++; 08. } 09.}
上面代码可行的原因是vector::erase() 返回一个新的迭代器,指向被删除元素的后面的元素。可以继续使用新的迭代器。
而出于某种未知的原因(multi)map::erase(), (multi)set::erase()没有返回这样的迭代器。(从C++11开始也支持返回迭代器了).
但是对于vector,诸如在0到99个数中删除所有奇数的问题,可以使用STL的remove(),remove_if()优化性能。代码如下:
01.bool isOdd(int value) 02.{ 03. return (value % 2) == 1; 04.} 05. 06.my_container.erase( std::remove_if(m_container.begin(), m_container.end(), isOdd), m_container.end());
让我们再看看不使用remove_if()的版本:
01.for (it != my_container.end();/**blank*/ ) { 02. if (*it % 2 == 1) { 03. it = my_container.erase(it); 04. } 05. else{ 06. it++; 07. } 08. }
如果你阅读过erase()源码或了解它是如何工作的,性能问题就显而易见:erase()删除一个元素的操作是被删除元素后面的所有元素依次
向前移动一个元素的位置,然后删除最后一个元素,时间复杂度为O(n^2)。
remove(),remove_if()的时间复杂度为O(n),删除元素的操作如下所示:
01.template<class ForwardIt, class UnaryPredicate> 02.ForwardIt remove_if(ForwardIt first, ForwardIt last, UnaryPredicate p) 03.{ 04. ForwardIt result = first; 05. for (; first != last; ++first) { 06. if (!p(*first)) { 07. *result++ = *first; 08. } 09. } 10. return result; 11. }
从前向后遍历容器所有元素,将待保留的元素向前移动,占据待删除的元素的位置,remove_if()返回新的元素范围(begin,end)中的end,
记为new_end_of_range ,再调用erase()删除从new_end_of_range到my_container.end()之间的所有元素。
实际上,remove_if() 没有删除容器中的任何元素,它没有改变my_container.end(), 调用remove_if()后容器元素个数不会改变!!删除元素的工作
交给了erase().
Scott Meyers在他的”Effective STL”中关于此问题的讨论中也使用了remove_if(),由此看来,他的确是提出了一些让STL effective的建议。
深入学习STL迭代器失效问题:
在google中搜索 stl iterator invalidation rules 可以获得很多有关STL迭代器失效的有关内容。
References:
1. STL remove_if() http://en.cppreference.com/w/cpp/algorithm/remove
2.More C++ Idioms/Erase-Remove http://en.wikibooks.org/wiki/More_C%2B%2B_Idioms/Erase-Remove
3.Effective STL, Item 32 - Scott Meyers
4.Cpp Invalid Iterators [对各种迭代器失效的情况进行了讲解分类]
http://www.angelikalanger.com/Conferences/Slides/CppInvalidIterators-DevConnections-2002.pdf
5.以下是stackoverflow上关于在迭代时删除容器中元素的讨论:
http://stackoverflow.com/questions/1604588/iterate-vector-remove-certain-items-as-i-go
http://stackoverflow.com/questions/3747691/stdvector-iterator-invalidation?rq=1
http://stackoverflow.com/questions/2874441/deleting-elements-from-stl-set-while-iterating?rq=1
http://stackoverflow.com/questions/1038708/erase-remove-contents-from-the-map-or-any-other-stl
-container-while-iterating/1038761#1038761
http://stackoverflow.com/questions/799314/difference-between-erase-and-remove?rq=1
-------------------------------------------------------------------------------------------------------------------------------------------------------------
http://hi.baidu.com/ewook/item/53e11ec2a5bea42c47d5c065
http://blog.chinaunix.net/uid-26335251-id-3347483.html
http://www.cnblogs.com/yc_sunniwell/archive/2010/06/25/1764934.html