我想使用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;
}
这段代码显然崩溃了,因为我在迭代它时改变了向量的结尾。实现这一目标的最佳方法是什么?即有没有办法做到这一点,而不是多次迭代矢量或创建一个矢量的副本?
答案 0 :(得分:144)
std::vector<int>& vec = myNumbers; // use shorter name
vec.erase(std::remove(vec.begin(), vec.end(), number_in), vec.end());
remove
会在number_in
的开头压缩与要删除的值(vector
)不同的元素,并将迭代器返回到第一个元素之后范围。然后erase
删除这些元素(其值未指定)。
答案 1 :(得分: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());
答案 2 :(得分:12)
您可以使用索引访问进行迭代,
避免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
方法,但可以解决您的任务。
使用pure 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());
}
答案 3 :(得分:3)
根据你这样做的原因,使用std::set可能比std :: vector更好。
它允许每个元素只出现一次。如果多次添加,则无论如何都只会删除一个实例。这将使擦除操作变得微不足道。 擦除操作的时间复杂度也低于向量,但是,在集合上添加元素的速度较慢,因此可能没什么优势。
如果您对向量中添加元素的次数或元素的添加顺序感兴趣,这当然不会起作用。
答案 4 :(得分:0)
要擦除第一个元素,您可以使用:
vector<int> mV{ 1, 2, 3, 4, 5 };
vector<int>::iterator it;
it = mV.begin();
mV.erase(it);