大家注意vector, list, set, map成員函數(shù)erase
#include <iostream>
#include <vector>
#include <list>
#include <iterator>
using namespace std;
void Remove1(vector<int> &vec, int num)
{
vector<int>::iterator iter;
for (iter=vec.begin(); iter!=vec.end(); ++iter)
{
if (*iter == num)
{
vec.erase(iter);
}
}
}
void Remove2(list<int> &lst, int num)
{
list<int>::iterator iter;
for (iter=lst.begin(); iter!=lst.end(); ++iter)
{
if (*iter == num)
{
lst.erase(iter);
}
}
}
int main(void)
{
int arr[] = {1, 3, 5, 5, 7, 9};
int num = sizeof(arr) / sizeof(arr[0]);
vector<int> vec(arr, arr+num);
list<int> lst(arr, arr+num);
Remove1(vec, 5);
copy(vec.begin(), vec.end(), ostream_iterator<int>(cout, " "));
//Remove2(lst, 5);
//copy(lst.begin(), lst.end(), ostream_iterator<int>(cout, " "));
return 0;
}
請(qǐng)看上面的Remove1、Remove2這兩個(gè)函數(shù)是刪除容器中所有為num的元素,乍一看好像沒(méi)什么問(wèn)題,其實(shí)是錯(cuò)的。
它們編譯都沒(méi)什么問(wèn)題,但是運(yùn)行有問(wèn)題:
先注釋Remove2這兩行, 編譯運(yùn)行, 得到結(jié)果是:1 3 5 7 9
顯然結(jié)果是錯(cuò)誤的,因?yàn)檫€有個(gè)5沒(méi)刪掉。造成這樣結(jié)果的原因是vec.erase之后iter會(huì)自加,執(zhí)行下次for循環(huán)時(shí)++iter,這樣就跳過(guò)了與所刪除元素相鄰的元素了。
再來(lái)看下Remove2, 注釋掉Remove1, 刪除Remove2的注釋。
運(yùn)行出錯(cuò),vs2005 Expression:list iterator not incrementable,說(shuō)明list迭代器++iter有問(wèn)題。
list是一個(gè)雙向鏈表,在erase(iter)之后iter失效了,printf("0x%x", iter);得到結(jié)果為0,只有erase的返回值才能指向下一個(gè)元素。
Remove1的正確用法是:vec.erase(remove(vec.begin(), vec.end(), 5), vec.end());
Remove2的正確用法是:lst.remove(5);
下面是另一種正確寫法:
void Remove1(std::vector<int> &vec, int num)
{
std::vector<int>::iterator iter = vec.begin();
while (iter != vec.end())
{
if (*iter == num)
{
iter = vec.erase(iter);
}
else
{
++iter;
}
}
}
// 刪除某一類元素, 比如:偶數(shù)
void Remove3(std::set<int>& sets)
{
std::set<int>::iterator iter = sets.begin();
while (iter != sets.end())
{
if (0 == (*iter)%2)
{
// 注意這里不能寫成++iter,后面說(shuō)明原因
sets.erase(iter++);
}
else
{
++iter;
}
}
}
void Remove4(std::map<int, int>& maps)
{
std::map<int, int>::iterator iter = maps.begin();
while (iter != maps.end())
{
if (0 == (iter->first)%2)
{
maps.erase(iter++);
}
else
{
++iter;
}
}
}
set和map是由紅黑樹(shù)來(lái)實(shí)現(xiàn)的,當(dāng)erase的時(shí)候迭代器就失效了,也就是說(shuō)我們要在迭代器失效之前保留一個(gè)副本,根據(jù)這個(gè)副本我們才能繼續(xù)遍歷下一個(gè)元素。i++和++i很明顯前者符合我們的要求所以在erase里面是iter++
相關(guān)文章

C語(yǔ)言詳解鏈?zhǔn)疥?duì)列與循環(huán)隊(duì)列的實(shí)現(xiàn)

64位linux 編譯c提示gnu/stubs-32.h:No such file or directory的解決方法

Mac OS X 10.8 中編譯APUE(Unix環(huán)境高級(jí)編程)的源代碼過(guò)程

教你如何使用qt quick-PathView實(shí)現(xiàn)好看的home界面

C語(yǔ)言數(shù)據(jù)結(jié)構(gòu)通關(guān)時(shí)間復(fù)雜度和空間復(fù)雜度