天天看點

std::map循環周遊删除元素出現詭異現象

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函數:

std::map循環周遊删除元素出現詭異現象

可以看到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";
           

注:不可以是前置++

STL