今天我发现一个被骗好多年的事情,直接看代码
void remove() {
std::vector<int> vec;
vec.push_back(1);
vec.push_back(2);
for(auto it=vec.begin();it!=vec.end();) {
if(*it==1) {
vec.erase(it++);
} else {
it++;
}
}
}
以上代码是网上常见的vector遍历删除的例子,貌似也可以工作,我也这么认为了好多年!现在我们稍加改动一下!
void remove() {
std::vector<int> vec;
vec.push_back(1);
vec.push_back(1);
vec.push_back(2);
for(auto it=vec.begin();it!=vec.end();) {
if(*it==1) {
vec.erase(it++);
} else {
it++;
}
}
}
你会发现这次有点不对,并不是所有的1都被删除了,为什么会这样呢?原因就是下面这句代码:
vec.erase(it++);
原来vector执行删除后,后面的成员会前移,也就是it并没有变,但是他已经指向下一个元素了。这样执行it++后,相当于it指向了下下个元素,也就是说把下一个1给跳过去了。那怎么解决呢?一般来说我们把++去掉就行了,但是这样美感不太好,所以我们需要这样写
void remove() {
std::vector<int> vec;
vec.push_back(1);
vec.push_back(1);
vec.push_back(2);
for(auto it=vec.begin();it!=vec.end();) {
if(*it==1) {
it=vec.erase(it);
} else {
it++;
}
}
}
回顾:
既然有问题,那这块深入人心的代码是怎么流传的呢?除了稍微隐蔽一点,最大的原因是,这段代码在list上是可以正常工作的,因为list删除时,不会产生成员移动的问题。