c++ - 从 vector 中选择特定元素

Question

我有一个 vector v1 和一个相同大小的 bool vector v2。我想从 v1 中删除所有值，使得 v2 的并行元素为 false:

vector<int> v3; // assume v1 is vector<int>
for (size_t i=0; i<v1.size(); i++)
    if (v2[i])
        v3.push_back(v1[i]);
v1=v3;

有没有更好的方法？

• 在 C++03 中
• 在 C++11 中

Answer 1

size_t last = 0;
for (size_t i = 0; i < v1.size(); i++) {
  if (v2[i]) {
    v1[last++] = v1[i];
  }
}
v1.erase(v1.begin() + last, v1.end());

本质上与您的相同，只是它可以就地工作，不需要额外的存储空间。这基本上是对 std::remove_if 的重新实现(这很难直接使用，因为它使用的函数对象被赋予了一个值，而不是容器中的索引或迭代器)。