向量迭代器根据条件擦除两个元素
Vector Iterator erase two elements on condition
如果满足某些条件,我目前正在尝试从向量中删除 2 个元素。我可以成功删除单个元素而不会发生 "vector iterator not dereferencable" 错误,我知道问题是由一次删除两个元素引起的,这会扰乱迭代器,但我不确定删除多个元素的正确方法立刻。
vector<SomeObj*> objs;
vector<SomeObj*>::iterator it = objs.begin();
while (it != objs.end())
{
vector<SomeObj*>::iterator it2 = objs.begin();
bool deleted = 0;
while (it2 != objs.end())
{
if ((*it)->somecondition(**it2))
{
delete *it2;
*it2 = NULL;
it = objs.erase(it2);
delete *it;
*it = NULL;
it = objs.erase(it); //Will error here due to invalidating the iterator
deleted = 1;
break;
}
++it2;
}
if (!deleted)
++it;
}
如果要改变容器,使用嵌套迭代器会很棘手。
我整理了一些示例代码来满足您的需求。我正在做的是通过将要删除的元素设置为 nullptr
来延迟删除,然后在循环中遇到它们时将其删除。
#include <iostream>
#include <vector>
class Example
{
public:
Example(int size) : size(size) {}
bool somecondition(const Example& other) const
{
return size == other.size;
}
int size;
};
int main()
{
std::vector<Example*> vec;
vec.push_back(new Example(1));
vec.push_back(new Example(2));
vec.push_back(new Example(3));
vec.push_back(new Example(2));
for (auto it1 = vec.begin(); it1 != vec.end();)
{
if (!*it1)
{
it1 = vec.erase(it1);
continue;
}
for (auto it2 = vec.begin(); it2 != vec.end(); ++it2)
{
if (!*it2)
{
vec.erase(it2);
// we need to start the outer loop again since we've invalidated its iterator
it1 = vec.begin();
break;
}
if (it1 != it2 && (*it1)->somecondition(**it2))
{
delete *it1;
*it1 = nullptr;
delete *it2;
*it2 = nullptr;
break;
}
}
++it1;
}
for (auto example : vec)
{
std::cout << example->size << std::endl;
}
return 0;
}
问题是第一次调用 erase() 很可能会使另一个迭代器无效。请参阅此 post 以快速了解在各种容器中无效的内容。我会说最简单的解决方案是首先遍历容器并标记要删除的条目但不要删除它们,然后在第二次扫描中只删除所有标记的内容。出于第二次扫描的性能原因,您应该使用 std::remove_if 或使用反向迭代器。
如果满足某些条件,我目前正在尝试从向量中删除 2 个元素。我可以成功删除单个元素而不会发生 "vector iterator not dereferencable" 错误,我知道问题是由一次删除两个元素引起的,这会扰乱迭代器,但我不确定删除多个元素的正确方法立刻。
vector<SomeObj*> objs;
vector<SomeObj*>::iterator it = objs.begin();
while (it != objs.end())
{
vector<SomeObj*>::iterator it2 = objs.begin();
bool deleted = 0;
while (it2 != objs.end())
{
if ((*it)->somecondition(**it2))
{
delete *it2;
*it2 = NULL;
it = objs.erase(it2);
delete *it;
*it = NULL;
it = objs.erase(it); //Will error here due to invalidating the iterator
deleted = 1;
break;
}
++it2;
}
if (!deleted)
++it;
}
如果要改变容器,使用嵌套迭代器会很棘手。
我整理了一些示例代码来满足您的需求。我正在做的是通过将要删除的元素设置为 nullptr
来延迟删除,然后在循环中遇到它们时将其删除。
#include <iostream>
#include <vector>
class Example
{
public:
Example(int size) : size(size) {}
bool somecondition(const Example& other) const
{
return size == other.size;
}
int size;
};
int main()
{
std::vector<Example*> vec;
vec.push_back(new Example(1));
vec.push_back(new Example(2));
vec.push_back(new Example(3));
vec.push_back(new Example(2));
for (auto it1 = vec.begin(); it1 != vec.end();)
{
if (!*it1)
{
it1 = vec.erase(it1);
continue;
}
for (auto it2 = vec.begin(); it2 != vec.end(); ++it2)
{
if (!*it2)
{
vec.erase(it2);
// we need to start the outer loop again since we've invalidated its iterator
it1 = vec.begin();
break;
}
if (it1 != it2 && (*it1)->somecondition(**it2))
{
delete *it1;
*it1 = nullptr;
delete *it2;
*it2 = nullptr;
break;
}
}
++it1;
}
for (auto example : vec)
{
std::cout << example->size << std::endl;
}
return 0;
}
问题是第一次调用 erase() 很可能会使另一个迭代器无效。请参阅此 post 以快速了解在各种容器中无效的内容。我会说最简单的解决方案是首先遍历容器并标记要删除的条目但不要删除它们,然后在第二次扫描中只删除所有标记的内容。出于第二次扫描的性能原因,您应该使用 std::remove_if 或使用反向迭代器。