给定迭代器列表，如何从向量中删除元素？

Question

我有一个 int 向量和一个映射，其中包含一些指向该向量的迭代器作为值。我需要从地图中删除键，以及值指向的向量元素。我的代码看起来像这样：

using RenderData = int;
using Element = std::string;

struct Ref {
    std::vector<RenderData>::iterator ref;
    std::function<int()> update;
    bool should_remove;
};

int main() {
    std::vector<RenderData> ints{1, 2, 3, 4, 5, 6, 7, 8, 9};
    std::unordered_map<Element, Ref> elements;

    // Here, I need to remove some elements, and their associated number
}

我实现了一个类似于this one 的erase_if 函数。

所以我的初始代码如下所示：

erase_if(elements, [&](auto&& element) {
    if (element.second.should_remove) {
        ints.erase(element.second.ref);
        return true;
    }

    return false;
});

显然没有用。擦除元素会使其他迭代器指向错误的对象，并且在某些情况下超出范围。所以我尝试了：

std::vector<std::vector<RenderData>::iterator> to_remove;

erase_if(elements, [&](auto&& element) {
    // condition based on the string content
    if (element.second.should_remove) {
        to_remove.emplace_back(element.second.ref);
        return true;
    }

    return false;
});

// Sort in descending order
std::sort(to_remove.begin(), to_remove.end(), std::greater<>{});

// stuff

for (auto&& it : to_remove) {
    ints.erase(it); // nothing can go wrong right?
}

再一次，我有时会擦除错误的元素。

给定存储在某个地图中的迭代器，我可以从向量中删除迭代器指向的元素吗？

---

更新：

似乎在上一个 sn-p 中我交换了向量中的一些元素，从而擦除了错误的元素。现在它似乎工作了，但我仍然很好奇我们可以用什么方法从迭代器列表中删除向量中的元素。

Answer 1

使用迭代器版本。 注意：

1. 迭代器只有在重新分配时才会失效。
2. 仅用索引替换迭代器不会改变从向量中删除某些值时它们无效的事实。
3. 使用索引仍然是一种更好的方法，因为以后添加更多元素时它们不会失效。

方法： 使用不应删除的元素创建向量的副本

using Element = std::string;
using RenderData = int;
struct Ref {
    std::vector<RenderData>::iterator itr;
    bool should_remove;
};

struct Main {
    std::vector<RenderData> ints;
    std::unordered_map<Element, Ref> elements;
    void remove_stuff(){
        std::vector<RenderData> localCopy;        
        localCopy.swap(ints);
        ints.reserve(localCopy.size());        
        for(auto it = elements.begin(); it != elements.end();) {
            Ref& ref = it->second;
            if(ref.should_remove) {
                it = elements.erase(it);
            } else {
                ints.push_back(std::move(*ref.itr));
                ref.itr = ints.end() - 1;                
                it++;
            }
        }            
    }
};

链接：https://godbolt.org/g/SouZ5E

Answer 2

A 计划： 最好把std::vector换成std::list，std::list的迭代器在擦除后不会失效。

B计划：

std::vector<int> newOne;
copy_if(oldOne.begin(), oldPOne.end(), std::back_inserter(newOne), [](){ //to test not in the map ~~ });
oldOne = std::move(newOne)

Answer 3

使用索引而不是迭代器。替换

struct Ref {
    std::vector<RenderData>::iterator ref;
    std::function<int()> update;
    bool should_remove;
};
std::vector< std::vector<RenderData>::iterator> to_remove;

与

struct Ref {
    std::size_t ref;
    std::function<int()> update;
    bool should_remove;
};
std::vector< std::size_t> to_remove;

然后你对后代排序的好主意就会奏效。通过索引删除元素调用ints.erase(ints.begin() + ref)

迭代器的排序是一个错误的想法。向量迭代器的实现方式是不可预测的。它们的排序顺序与指针/地址/索引相同的可能性很小。

迭代器擦除（迭代器位置）； （直到 C++11）

迭代器擦除（const_iterator pos）； (C++11 起)

移除 pos 处的元素。

在擦除点或之后使迭代器和引用失效，包括 end() 迭代器。