std::map循环遍历删除元素出现诡异现象

	std::map 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循环遍历删除元素出现诡异现象_第1张图片

可以看到erase并没有返回值,不像vector一样会erase会有返回值,表示被删除元素下一个元素的位置。

分析上述诡异现象:当元素是2或者3的时候erase(it)会导致it迭代器失效,但是没有更新迭代器,而是继续继续判断else语句成立所以会继续执行++it,导致最终程序奔溃。

解决上述问题:如下代码

    std::map 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 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)