首页 文章

从矢量中删除元素

提问于
浏览 次
85

我想使用erase方法从向量中清除元素 . 但是这里的问题是元素不能保证在向量中只出现一次 . 它可能存在多次，我需要清除所有这些 . 我的代码是这样的：

void erase(std::vector<int>& myNumbers_in, int number_in)
{
    std::vector<int>::iterator iter = myNumbers_in.begin();
    std::vector<int>::iterator endIter = myNumbers_in.end();
    for(; iter != endIter; ++iter)
    {
        if(*iter == number_in)
        {
            myNumbers_in.erase(iter);
        }
    }
}

int main(int argc, char* argv[])
{
    std::vector<int> myNmbers;
    for(int i = 0; i < 2; ++i)
    {
        myNmbers.push_back(i);
        myNmbers.push_back(i);
    }

    erase(myNmbers, 1);

    return 0;
}

这段代码显然崩溃了，因为我在迭代它时改变了向量的结尾 . 实现这一目标的最佳方法是什么？即有没有办法做到这一点，而不是多次迭代矢量或创建一个矢量的副本？

c++ vector stl erase

4 回答

  • 3

    使用remove/erase idiom：

    std::vector<int>& vec = myNumbers; // use shorter name
vec.erase(std::remove(vec.begin(), vec.end(), number_in), vec.end());

    会发生什么是 remove 压缩与 vector 开头的要删除的值（ number_in ）不同的元素，并将迭代器返回到该范围之后的第一个元素 . 然后 erase 删除这些元素（谁的值未指定） .

    回复于
  • 47

    调用erase会使迭代器失效，你可以使用：

    void erase(std::vector<int>& myNumbers_in, int number_in)
{
    std::vector<int>::iterator iter = myNumbers_in.begin();
    while (iter != myNumbers_in.end())
    {
        if (*iter == number_in)
        {
            iter = myNumbers_in.erase(iter);
        }
        else
        {
           ++iter;
        }
    }

}

    或者您可以将std::remove_if与仿函数和std :: vector :: erase一起使用：

    struct Eraser
{
    Eraser(int number_in) : number_in(number_in) {}
    int number_in;
    bool operator()(int i) const
    {
        return i == number_in;
    }
};

std::vector<int> myNumbers;
myNumbers.erase(std::remove_if(myNumbers.begin(), myNumbers.end(), Eraser(number_in)), myNumbers.end());

    在这种情况下，您可以使用std::remove代替编写自己的仿函数：

    std::vector<int> myNumbers;
myNumbers.erase(std::remove(myNumbers.begin(), myNumbers.end(), number_in), myNumbers.end());
    回复于
  • 140

    • 您可以使用索引访问进行迭代，

    • 为了避免O（n ^ 2）复杂度，您可以使用两个索引，i - 当前测试索引，j - 索引来存储下一个项目，并在循环结束时使用矢量的新大小 .

    码：

    void erase(std::vector<int>& v, int num)
{
  size_t j = 0;
  for (size_t i = 0; i < v.size(); ++i) {
    if (v[i] != num) v[j++] = v[i];
  }
  // trim vector to new size
  v.resize(j);
}

    在这种情况下，您没有迭代器的无效，复杂性是O（n），并且代码非常简洁，您不需要编写一些辅助类，尽管在某些情况下使用辅助类可以在更灵活的代码中受益 .

    此代码不使用 erase 方法，但可以解决您的任务 .

    使用纯stl，您可以通过以下方式执行此操作（这与Motti的答案类似）：

    #include <algorithm>

void erase(std::vector<int>& v, int num) {
    vector<int>::iterator it = remove(v.begin(), v.end(), num);
    v.erase(it, v.end());
}
    回复于
  • 12

    根据你为什么这样做，使用std::set可能比std :: vector更好 .

    它允许每个元素只出现一次 . 如果多次添加，则无论如何都只会有一个实例要擦除 . 这将使擦除操作变得微不足道 . 擦除操作也将具有比向量更低的时间复杂度，然而，添加元素在集合上更慢，因此它可能没有多大优势 .

    如果您对向量中添加元素的次数或元素的添加顺序感兴趣，这当然不起作用 .

    回复于

相关问题