設(shè)計一段演示程序:打印n個數(shù)字,然后指定擦掉其中的某一個,重新打印剩下的數(shù)字。
#include <iostream>
#include <vector>
using std::cout;
using std::cin;
using std::endl;
using std::vector;
int main(int argc, char* argv[])
{
const int MAX = 10;
vector<int> v;
for (int i = 0; i < MAX; i++) {
v.push_back(i);
cout << v[i] << '\t';
}
int ers = -1;
while ( ers < 0 || ers >= MAX) {
cout << "Erase No.";
cin >> ers;
}
for (vector<int>::const_iterator i = v.begin(); i != v.end(); i++) {
if (*i == ers) {
v.erase(i);
}
}
for (vector<int>::const_iterator i = v.begin(); i != v.end(); i++) {
cout << *i << '\t';
}
cout << endl;
return 0;
}
這個程序看似沒什么錯誤,可以順利編譯,也可以順利運行,甚至,大部分情況下可可以正常完成。
我們這里設(shè)置的n為10(MAX),事實上,我們擦掉0到8都沒有問題,但是,我們要擦掉9的時候,程序卻出錯了!
在我們遍歷查找對等值的循環(huán)中,一開始v.end()指向第10個元素(數(shù)值為9)的后面一個位置(不存在的第11個元素的位置)。當(dāng)?shù)髦赶虻?0個元素(數(shù)值為9)的時候,v.erase()生效運行;下一輪循環(huán)中,迭代器本來應(yīng)該指向第11個元素的位置,并且等于v.end()并結(jié)束循環(huán)。但是,因為我們擦掉了vector中的一個元素,v.end()指向的是現(xiàn)在的最后一個元素——第9個元素的后面,也就是第10個元素的位置。這樣,迭代器到了11,而判斷確是其是否到10,這將永遠(yuǎn)無法實現(xiàn),形成了一個邏輯bug,所以系統(tǒng)拋出錯誤了。
一個修正的辦法是把需要擦掉的迭代器找出來,在循環(huán)結(jié)束后再擦掉,下面修改后的程序就可以正常的擦掉9了。
#include <iostream>
#include <vector>
using std::cout;
using std::cin;
using std::endl;
using std::vector;
int main(int argc, char* argv[])
{
const int MAX = 10;
vector<int> v;
for (int i = 0; i < MAX; i++) {
v.push_back(i);
cout << v[i] << '\t';
}
int ers = -1;
while ( ers < 0 || ers >= MAX) {
cout << "Erase No.";
cin >> ers;
}
vector<int>::const_iterator ers_i;
for (vector<int>::const_iterator i = v.begin(); i != v.end(); i++) {
if (*i == ers) {
ers_i = i;
}
}
v.erase(ers_i);
for (vector<int>::const_iterator i = v.begin(); i != v.end(); i++) {
cout << *i << '\t';
}
cout << endl;
return 0;
}