std::map<int, int,cmp> m1{ { 1, 1 }, { 3, 3 }, { 2, 2 } };
m1.insert(std::make_pair(4, 4));
for (auto it = m1.begin(); it != m1.end();)
{
if (it->first == 3 || it->first == 2)
m1.erase(it);
else
++it;
}
这段代码在运行的时候会出现if语句执行完继续执行else语句。为什么会出现这种现象?
先来看一下map中的erase函数:
可以看到erase并没有返回值,不像vector一样会erase会有返回值,表示被删除元素下一个元素的位置。
分析上述诡异现象:当元素是2或者3的时候erase(it)会导致it迭代器失效,但是没有更新迭代器,而是继续继续判断else语句成立所以会继续执行++it,导致最终程序奔溃。
解决上述问题:如下代码
std::map<int, int,cmp> m1{ { 1, 1 }, { 3, 3 }, { 2, 2 } };
m1.insert(std::make_pair(4, 4));
for (auto it = m1.begin(); it != m1.end();)
{
if (it->first == 3 || it->first == 2)
m1.erase(it++);
else
++it;
}
for (auto it : m1)
std::cout << it.first << " " << it.second << "\n";
在上述正确代码中erase参数是it++,因为it再给erase传参之后,在迭代器没有失效之前进行了++操作指向下一个元素,所以这种能够正确删除元素。或者可以定义一个临时变量。如下:
std::map<int, int,cmp> m1{ { 1, 1 }, { 3, 3 }, { 2, 2 } };
m1.insert(std::make_pair(4, 4));
for (auto it = m1.begin(); it != m1.end();)
{
auto tmp = it;
++tmp;
if (it->first == 3 || it->first == 2)
{
m1.erase(it);
it = tmp;
}
else
++it;
}
for (auto it : m1)
std::cout << it.first << " " << it.second << "\n";
注:不可以是前置++