c++ - 连续删除 vector 元素

Question

如果这些元素相等，我需要从 vector a 和 b 中删除元素。据我了解，当我删除一个元素时，其他元素也会移动。因此，a[1] 变为 a[0]、a[2] -- a[1] 等等。但是我的代码不能正常工作:

#include <iostream>
#include <vector>

using namespace std;

bool remove(vector<int>& a, vector<int>& b){

     for (size_t i = 0; i < a.size(); ++i )
     {

      while ( true )
      {
          std::vector<int>::iterator it;
          it = std::find(b.begin(), b.end(), a[i]);
          if ( it != b.end() )
          {
              i = 0; //because of moving elements
              b.erase(b.begin()+i); 
              a.erase(a.begin()+i);
              break; 
          }

      }
    }
    return true;
}

int main(){
    vector<int> a;
    vector<int> b;

    a.push_back(1);
    a.push_back(2);
    a.push_back(3);

    b.push_back(1);
    b.push_back(2);
    b.push_back(3);

    remove(a,b);

    return 0;
}

问题是 - 最后一个相等的元素保留下来，所以我无法删除它们。考虑到不相等的元素可能位于 vector 的中间，我该如何解决这个问题:a = 1, 2, 3; b = 1, 3, 3 -- vector a中的2和 vector b中的3不应该被删除？

Answer 1

As I understand, when I remove an element, other elements move.

部分正确。如果您删除一个元素，之后 的元素将被向下推以填充孔。如果删除第 5 个元素，则原为第 6 个元素的元素现在变为第 5 个。但是第 4 个元素仍然是第 4 个。这当然只有在您删除之后才是正确的。所以你的代码是:

i =  // the element from a
it = // the iterator pointing to an element from b

i = 0;
b.erase(b.begin()+i); // erase the 1st element from b
a.erase(a.begin()+i); // erase the 1st element from a

你做所有这些工作来找到一对相等的元素，然后立即丢弃它——删除两个 vector 的前面的元素。这就是为什么它不起作用。你想删除你找到的元素:

b.erase(it);
a.erase(a.begin() + i); // no i = 0

请注意，如果您想删除在另一个数组中具有匹配项的每个元素，则此修复不会削减它 - 如果您在 a 中有两个 1 > 但只有 b 中的一个你永远不会删除第二个。为此，我们需要利用 Erase-remove idiom :

for (std::vector<int>::iterator itA = a.begin(); itA != a.end(); /* nothing */)
{
    std::vector<int>::iterator itB = std::find(b.begin(), b.end(), *itA);
    if (itB == b.end()) {
       // done with this element, move on
       ++itA;
    }
    else {
        // erase EVERYTHING
        b.erase(std::remove(itB, b.end(), *itA), b.end());
        itA = a.erase(std::remove(itA, a.end(), *itA), a.end());
    }
}