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";
注:不可以是前置++