使用迭代器從map或vector中刪除元素

std:map的迭代器:刪除時隻影響當前元素
#include "stdafx.h"
#include <map>

std::map<int, int> m;

int main()
{
	m[1] = 10;
	m[2] = 100;
	m[3] = 1000;

	std::map<int, int>::iterator it;
	std::map<int, int>::iterator itt;

	printf("1 m.len = %d\n", m.size());
	for (it = m.begin(); it != m.end();)
	{
		if (it->second == 100)
		{
			/////標準寫法/////
			m.erase(it++);
			
			/////等效寫法/////
			// itt = it;
			// it++;
			// m.erase(itt);
			
			/////錯誤寫法/////
			// itt = it;
			// m.erase(itt);
			// it++;
		}
		else
		{
			it++;
		}
	}
	printf("2 m.len = %d\n", m.size());
	return 0;
}

【原因分析】
it++的操作可拆分爲三個步驟:

  1. 首先把 it 備份一下。
  2. 把 it 加上1。
  3. 返回第一步備份的 it 。

所以標準寫法中執行m.erase(it++);語句,在執行erase之前,it 已經被加1了。erase會使得以前那個未被加一的 it 失效,而加了一之後的新的 it 是有效的。

等效寫法將標準寫法進行了拆分。

m.erase(it); // erase以後,it 已經失效,不能再用 it++;
那麼erase操作直接讓 it 失效,對失效的 it 進行 ++ 操作也是失效的。

std::vector的迭代器,刪除時自動返回下一個元素的位置,所以erase時要取出返回值。
#include <vector>

std::vector<int> m;

int main()
{
	m.push_back(10);
	m.push_back(100);
	m.push_back(1000);

	std::vector<int>::iterator it;
	std::vector<int>::iterator itt;

	printf("1 m.len = %d\n", m.size());
	for (it = m.begin(); it != m.end();)
	{
		printf("==> %d\n", *it);
		if (*it == 100)
		{
			it = m.erase(it);
			/////寫成下面這樣,效果也一樣/////
			// it = m.erase(it++);
		}
		else
		{
			it++;
		}
	}
	printf("2 m.len = %d\n", m.size());
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章