使用 for 循环擦除矢量中的元素

Erase elements in vector using for loop

本文关键字:元素 for 循环 擦除 使用      更新时间:2023-10-16
如何使用

for 循环通过其索引从向量中删除元素?我得到一个超出范围的矢量错误。我在下面有一个示例代码。

 vector<int> to_erase = {0, 1, 2}; 
 vector<int> data     = {3, 3, 3, 3};
 for(int i = 0; i < to_erase.size(); i++) {
    data.erase(data.begin() + to_erase[i]);
 }

我认为这是因为我的向量的大小在每次迭代中都会减小,因此它无法访问索引 2。

您通常会使用擦除-删除习惯用法来有效地从向量中删除多个元素(逐个擦除它们通常效率较低,而且如您所见,并不总是微不足道的)。在最一般的形式中,成语看起来像这样:

data.erase(remove_algorithm(begin(data), end(data)), end(data));

在您的情况下,remove_algorithm基于另一个向量中的索引,因此我们也需要提供这些索引:

data.erase(
    remove_indices(begin(data), end(data), begin(to_erase), end(to_erase)),
    end(data));

不幸的是,这样的算法不包含在标准库中。但是,给自己写1是微不足道的

template <typename It, typename It2>
auto remove_indices(It begin, It end, It2 idx_b, It2 idx_e) -> It {
    using idx_t = typename std::iterator_traits<It2>::value_type;
    std::sort(idx_b, idx_e, std::greater<idx_t>{});
    for (; idx_b != idx_e; ++idx_b) {
        auto pos = begin + *idx_b;
        std::move(std::next(pos), end--, pos);
    }
    return end;
}

在这里,我们首先从大到小对要删除的索引进行排序。接下来,我们遍历这些索引。然后,我们(最有效地)将当前位置(要删除)和向量末端之间的所有元素向前移动一个。随后,结尾减少 1(以解释元素被删除的事实)。

实时代码


1 *咳咳*一旦你删除了代码中所有愚蠢的错别字。

我认为这是因为我的向量的大小在每次迭代中都会减小

是的!

您可以通过保留一个额外的变量来计算删除的元素,如下所示:

#include <iostream>
#include <vector>
using namespace std;
int main() {
        vector<int> to_erase = {0, 1, 2}; 
        vector<int> data     = {3, 3, 3, 3};
        int count_removed = 0;
        for(unsigned int i = 0; i < to_erase.size(); i++)
                data.erase(data.begin() + to_erase[i] - count_removed++);
        for(unsigned int i = 0; i < data.size(); ++i)
                        cout << data[i] << "n";
        return 0;
}

输出:

3

当我第一次使用std::erase()时,我遇到了同样的问题,好问题,+1。

在迭代的同时删除元素集合是不安全的,而且成本高昂。我建议每个符合您标准的元素在最后都与一个元素交换。(最后,因为从最后擦除会更便宜。您可以跟踪您从向量末尾返回了多少(基于交换的数量),并尽早打破我们的循环。现在,根据您交换的元素数量,您可以执行以下操作:

data.resize(data.size() - reverse_counter);

int pos = data.size() - reverse_counter;
data.erease(data.begin()+pos, data.end();

sudo 代码只是为了解释这个想法。

如参考文献中所述,擦除不最终会导致重新分配,这是昂贵的。值得记住的事情:http://www.cplusplus.com/reference/vector/vector/erase/

我认为

这是一个糟糕的设计,因为您将更改 for 循环不变性,并且需要很多解决方法才能实现这一目标。无论如何,如果你真的想使用 for 循环,你可以标记你想要删除的内容并运行一个 stl remove_if,如下所示:

#include <iostream>
#include <vector>
#include <limits>
#include <algorithm>
using namespace std;
int main() {
    vector<int> to_erase = {0, 1, 2}; 
    vector<int> data     = {3, 3, 3, 3};
    cout << "Before:n" ;
    for(int i=0; i<data.size(); i++)
        cout << i << "t";
    cout << endl;   
    for(int i=0; i<data.size(); i++)
        cout << data[i] << "t";
    cout << endl;
    for(int i = 0; i < to_erase.size(); i++) {
        //data.erase(data.begin() + to_erase[i]);
        data[i] = numeric_limits<int>::max();
    }
    data.erase(remove_if(data.begin(), 
                         data.end(), 
                         [](int i){return i==numeric_limits<int>::max();}), data.end());
    cout << "Next:n" ;
    for(int i=0; i<data.size(); i++)
        cout << i << "t";
    cout << endl;
    for(int i=0; i<data.size(); i++)
        cout << data[i] << "t";
    return 0;
}