【发布时间】:2011-02-21 21:34:17
【问题描述】:
我需要检查一组并删除符合预定义条件的元素。
这是我写的测试代码:
#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 内有多个测试条件,则每个条件都必须递增迭代器。我更喜欢这段代码,因为迭代器只在一个地方递增,使代码不易出错且更具可读性。
【问题讨论】:
-
实际上,我在问我之前阅读了这个问题(和其他问题),但由于它们与其他 STL 容器有关,而且我的初始测试显然有效,我认为它们之间存在一些差异。只有在马特的回答之后,我才想到使用 valgrind。尽管如此,我更喜欢我的新解决方案而不是其他解决方案,因为它通过仅在一个地方增加迭代器来减少出错的机会。谢谢大家的帮助!
-
@pedromanoel
++it应该比it++更有效,因为它不需要使用不可见的迭代器临时副本。 Kornel 的版本虽然更长,但可以确保最有效地迭代未过滤的元素。 -
@Alnitak 我没有考虑过这一点,但我认为性能差异不会那么大。该副本也在他的版本中创建,但仅适用于匹配的元素。所以优化程度完全取决于集合的结构。在相当长的一段时间里,我预先优化了代码,在这个过程中损害了可读性和编码速度......所以我会在使用其他方式之前进行一些测试。
标签: c++ iterator set std c++-standard-library