我需要检查一组并删除符合预定义标准的元素。
这是我写的测试代码:
#include <set> #include <algorithm> void printElement(int value) { std::cout << value << " "; } int main() { int initNum[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; std::set<int> numbers(initNum, initNum + 10); // print '0 1 2 3 4 5 6 7 8 9' std::for_each(numbers.begin(), numbers.end(), printElement); std::set<int>::iterator it = numbers.begin(); // iterate through the set and erase all even numbers for (; it != numbers.end(); ++it) { int n = *it; if (n % 2 == 0) { // wouldn't invalidate the iterator? numbers.erase(it); } } // print '1 3 5 7 9' std::for_each(numbers.begin(), numbers.end(), printElement); return 0; }
起初,我认为在迭代时从集合中删除一个元素会使迭代器无效,并且 for 循环中的增量将具有未定义的行为。尽管如此,我执行了这个测试代码并且一切顺利,我无法解释为什么。
我的问题: 这是标准集的定义行为还是此实现特定?顺便说一句,我在 ubuntu 10.04(32 位版本)上使用 gcc 4.3.3。
谢谢!
建议的解决方案:
这是从集合中迭代和擦除元素的正确方法吗?
while(it != numbers.end()) { int n = *it; if (n % 2 == 0) { // post-increment operator returns a copy, then increment numbers.erase(it++); } else { // pre-increment operator increments, then return ++it; } }
编辑:首选解决方案
我找到了一个对我来说似乎更优雅的解决方案,即使它完全一样。
while(it != numbers.end()) { // copy the current iterator then increment it std::set<int>::iterator current = it++; int n = *current; if (n % 2 == 0) { // don't invalidate iterator it, because it is already // pointing to the next element numbers.erase(current); } }
如果 while 内有多个测试条件,则每个测试条件都必须递增迭代器。我更喜欢这段代码,因为迭代器 只在一个地方 递增,使代码不易出错且更具可读性。
这取决于实现:
标准 23.1.2.8:
插入成员不应影响迭代器和对容器的引用的有效性,而擦除成员应仅使迭代器和对被擦除元素的引用无效。
也许你可以试试这个——这是符合标准的:
for (auto it = numbers.begin(); it != numbers.end(); ) { if (*it % 2 == 0) { numbers.erase(it++); } else { ++it; } }
请注意,it++ 是后缀,因此它将旧位置传递给擦除,但由于运算符的原因,它首先跳转到新位置。
2015.10.27 更新: C11 已解决该缺陷。iterator erase (const_iterator position);返回一个迭代器,指向被移除的最后一个元素之后的元素(或者set::end,如果最后一个元素被移除)。所以 C11 风格是:
iterator erase (const_iterator position);
set::end
for (auto it = numbers.begin(); it != numbers.end(); ) { if (*it % 2 == 0) { it = numbers.erase(it); } else { ++it; } }