如何删除与另一个向量中的某些元素匹配的向量中的元素

Question

我想删除向量v2 中与另一个向量v1 中的元素匹配的元素，其中v1 中的每个元素在v2 中都有对应的元素，但反之则不然。

我的想法是像这样使用std::vector::erase、std::remove_if 和std::find：

std::vector<int> v1{ 6, 2, 4 };
std::vector<int> v2{ 8, 2, 9, 4, 5, 3, 7, 1, 6 };

v2.erase(
  std::remove_if(v2.begin(), v2.end()
  , [&v1](auto& item)
  {
    return std::find(v1.begin(), v1.end(), item) != v1.end();
  })
);

哪个有效。但是，我想知道我是否可以不匹配已经匹配的那些，因为这是从v1 -> v2 开始时的 1:1 关系。 v1之后会被吹走，所以可以修改。

Answer 1

排序 v1。使用 std 相等范围查找其中的元素。将它与您的 remove-erase 习语代码一起使用（带有错误修复）。

这是 O((M+N)lg N)

std::sort( v1.begin(), v1.end() );
auto removed=std::remove_if(
  v2.begin(), v2.end(), 
  [&v1](auto& item)
  {
    auto r= std::equal_range(v1.begin(), v1.end(), item);
    return r.first!=r.second;
  }
);
v2.erase(removed, v2.end());

请注意，您在擦除调用中忘记了.end，这是执行 remove-erase 时常见的严重错误。

如果条目数量合理，则编辑源向量以消除元素是不值得的，除非是非常花哨的数据结构。

Answer 2

您可以实现线性复杂性来解决您的问题，但需要额外的内存。您可以将第一个向量转移到散列（unordered_set）。因此，您将获得 O(N) 复杂度（不包括创建哈希的努力）

std::vector<int> v1{ 6, 2, 4 };
std::vector<int> v2{ 8, 2, 9, 4, 5, 3, 7, 1, 6 };

std::unordered_set<int> hash_v1( v1.begin(), v1.end() );

// linear complexity for remove_if
v2.erase( std::remove_if( v2.begin(), v2.end(), [&hash_v1] ( auto& item )
{
    return hash_v1.find( item ) != hash_v1.end(); // constant time
}), v2.end());

不要忘记 v2.end() 作为 ease 方法的第二个参数。

Answer 3

如果您被允许修改v1，那么类似以下的内容将起作用：

v2.erase(
  std::remove_if(v2.begin(), v2.end(),
    [&v1](auto& item) {
      auto it = std::find(v1.begin(), v1.end(), item);
      if (it != v1.end()) {
        v1.erase(it);
        return true;
      }
      return false;
    }
  ), v2.end());

编辑 1：首先对 v1 进行排序，然后使用 std::lower_bound 来获得 O(log n) 的搜索速度，而不是 O(n)。

编辑 2： 将 v1 的元素存储到 std::unordered_set 中以进行擦除并找到 O(1) 摊销。

编辑 3： 添加了 Yakk 提到的关于 erase 的错误修正。