迭代时从std :: set删除元素

问题描述 投票:135回答:8

我需要遍历一组并删除符合预定义条件的元素。

这是我编写的测试代码:

#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循环的增量将具有未定义的行为。即使我执行了此测试代码,但一切顺利,并且我无法解释原因。

我的问题:这是std集的已定义行为还是此实现特定?顺便说一下,我正在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);
    }
}

如果一段时间内有多个测试条件,则每个测试条件必须增加迭代器。我更喜欢此代码,因为迭代器递增了[[仅在一个位置,从而使代码不易出错且更具可读性。

c++ iterator set std c++-standard-library
8个回答
164
投票
这取决于实现:

18
投票
如果通过valgrind运行程序,则会看到许多读取错误。换句话说,是的,迭代器已失效,但是您在示例中很幸运(或者很倒霉,因为您没有看到未定义行为的负面影响)。一种解决方案是创建一个临时迭代器,增加温度,删除目标迭代器,然后将目标设置为温度。例如,重新编写循环,如下所示:

7
投票
您误解了“未定义行为”的含义。未定义的行为并不表示“如果执行此操作,则程序

will


2
投票
只是警告一下,在使用双端队列容器的情况下,所有检查双端队列迭代器与numbers.end()相等性的解决方案都可能在gcc 4.8.4上失败。即,删除双端队列的元素通常会使指向numbers.end()的指针无效:

1
投票
此行为是特定于实现的。为了保证迭代器的正确性,您应该使用“ it = numbers.erase(it);”声明是否需要删除元素,而在其他情况下只需增加迭代器。

1
投票
我认为在尝试删除由迭代器包装的对象时,使用STL方法'remove_if'可以帮助防止某些奇怪的问题。

1
投票
C ++ 20将具有“统一容器擦除”,您将能够编写:

0
投票
我遇到了同样的旧问题,并在下面的代码中发现了更多的[[understandable
© www.soinside.com 2019 - 2024. All rights reserved.