删除向量元素使用向量<bool>中的条件

Question

我有两个矢量a和b,大小相同.

vector<int>  a{    4,    3,   1,    3,   1};
vector<bool> b{false,false,true,false,true};

a如果b(同一索引)中的相同元素为真,我想删除元素.

应用函数后:a = 4,3,3

注意:我想使用std算法或函数而不是简单的for循环.

Answer 1

  std::vector<int> v {1,2,3,4,5,6};
  std::vector<bool> b {true, false, true, false, true, false};

  v.erase(std::remove_if(v.begin(), v.end(), 
      [&b, &v](int const &i) { return b.at(&i - v.data()); }), v.end());

现场演示

Answer 2

void filter(std::vector<int>& v, const std::vector<bool>& b)
{
    assert(v.size() == b.size());
    auto it = b.begin();
    v.erase(std::remove_if(v.begin(), v.end(), [&](int) { return *it++; }), v.end());
}

演示版