C++|迭代器失效问题

谈谈迭代器失效问题

我们先来看看一个很经典的问题。

//在一个vector<int> vc中，删除所有target元素
void func1(vector<int>& vc, int target) {
	for (auto it = vc.begin(); it != vc.end(); it++) {
		if (*it == target) {
			it = vc.erase(it);
		}
	}
}
int main() {
	vector<int> vc = { 3,3,2,3,3,3,5,3,6 };
	func1(vc, 3);
	for (auto i : vc)cout << i << " ";
}

运行结果：

可以发现，等于3的元素删除不干净，为什么呢？
因为在vector中，当erase()的时候，删除元素后会返回迭代器，而后面的元素往前移动一个单位，如果单纯it++则会错误。
我们通过下面这个程序可以来验证。

int main() {
	vector<int> v = { 1,2,3,4,5};
	auto it = v.begin();
	cout << "erase之前：" << &it << " -> " << *it << endl;
	it = v.erase(it);
	cout << "erase之后：" << &it << " -> " << *it << endl;
    return 0;
}

结果如下：

我们可以发现，删除前跟删除后，迭代器it，地址是相同的005BF854，但是指向内容改变了，因为我们删了内容为1的那个元素，也就是说后面的元素往前移动。
所以，在第一个程序中，当我们删除之后，本来元素就全部往前移动一个单位后，我们此时的it指向的是原位置，但原位置上已经是下一个元素了，所以这时候再it++就会使得跳过一个位置了，也就出现了“当3重复出现的时候就会有漏网之鱼”这样的问题，可以理解吧。
下面说一下解决办法，没删除则it++，有删除则不it++，如下。

void func1(vector<int>& vc, int target){
    for(auto it = vc.begin();it!=vc.end();it){
        if(*it == target){
            it = vc.erase(it);
        }
        else{
            it++;
        }
    }
}

运行结果就符合预期了。

另一方面我们来说说对关联容器map（同unordered_map、set、unordered_set）中，当erase()的时候，删除元素后迭代器就指向一个错误的地方，所以下一次引用这个迭代器时就会错误。
对于map与vector不同的是，map在erase()之后，迭代器就会失效了（真正的失效了，迭代器值不变，但是内容被释放了，此时it变成野指针，访问它相当于访问空指针）

int main() {
	unordered_map<int, int> m;
	int count = 0;
	for (int i : {3, 3, 2, 3, 3, 3, 5, 3, 6}) {
		m[count++] = i;
	}
	auto it = m.begin();
	cout << "erase之前：" << &it << " -> " << it->second << endl;
	m.erase(it);
	//cout << "erase之后：" << &it << " -> " << it->second << endl;
	//这个会访问越界，因为访问了一个已经释放的内存。
	cout << "erase之后：" << &it  << endl;
	return 0;
}

运行结果符合刚才的解释。
那么如果同样是单纯地erase()后直接it++，那肯定会引发错误然后奔溃。
解决的办法是：在erase()之前，用新的it_new = it++;保持it的同时，用临时变量 it_new去进行erase()。

void func2(map<int,int>&m, int target){
    for(auto it = m.begin();it != m.end();it){
        if(it->second == target){
            auto it_new = it++;	//这时候it已经移动下一位
            m.erase(it_new);	//临时变量it_new失效也无所谓了，有用的it已经跑到下一位了
        }
        else{
            it++;
        }
    }
}

这就是经典的迭代器失效问题。
一开始我在使用vector的时候，发现总是有删除不干净的情况，一步一步debug最后发现它就跳过去了，震惊之余赶紧学习并记录下来。

—————————————————————————————————
0726 参加字节跳动的面试，面试官大哥就出了这个问题，好在之前遇到过并认真去研究它的原因了哈哈哈哈哈哈，回来感谢自己。