在c++編程中,用到迭代器的時候,往往不知道如何刪除當前迭代器指向的元素。
erase函數:
返回下一個迭代器。
只使用vector的erase函數,記住,該函數是迭代器失效,返回下一個迭代器。
#include <iostream> #include <vector> using namespace std; int main() { vector<int> arr; arr.push_back(6); arr.push_back(7); arr.push_back(8); arr.push_back(9); for(vector<int>::iterator it=arr.begin(); it!=arr.end(); ) { if(* it == 8) { it = arr.erase(it); //不能寫成arr.erase(it); } else { ++it; }
看下面的一個程序,刪除值為4和7的元素,為什么只刪除了4?
#include<iostream> #include<string> #include<vector> using namespace std; int main() { vector<int> iVec; vector<int>::iterator it; int i; for( i=0;i<10;i++) iVec.push_back(i); for( i=0;i<iVec.size();i++) cout<<iVec[i]<<ends; cout<<endl; for(it=iVec.begin();it!=iVec.end();++it) { if(*it ==4 || *it == 7) { it=iVec.erase(it); } else ++it; } for( i=0;i<iVec.size();i++) cout<<iVec[i]<<ends; cout<<endl; }
沒有刪除7,為什么?
l例如上面這個 1 2 3 4 5 6 7 8 9
如果要刪除4 和7 兩個元素
則刪除4后iterator返回5,不滿足if()判斷,這時++it執行兩次,一次是else里面,一次是for()循環里面,導致7被錯過
改為:
it=iVec.erase(it);
--it;//這里回退一個
上面的程序基本上屬於弱智型程序,沒必要it++,使用下面的:
for(it=iVec.begin();it!=iVec.end();++it) { if(*it ==4 || *it == 7) { it=iVec.erase(it); } }
(2014-12-1修改:
上面程序經過網友指出if(*it ==4 || *it == 5)有問題,確實如此,2個相鄰的元素,在for++中被跳過了。為了處理這種情況,把for循環中的it++移動到for中。
for(it=iVec.begin();it!=iVec.end();)
{
if(*it==4 || *it==5)
it=iVec.erase(it);
else
it++;
}
這樣程序就可以處理4,5,4,7之類的了。)
不用循環結合remove使用:
1、remove並不是刪除,僅僅是移除,要加上erase才能完成刪除。
2、remove並不是刪除指定位置的元素,而移除所有指定的元素。
3、用algorithm代替成員函數不是一個好的選擇。
remove只是簡單地用"未刪除"對象來填補被刪除對象留下的缺口,每一個刪除對象在尾部還是會留下一個相應的"死亡"對象.
可以這樣來使用remove...
vector<int>vec;
vec.erase(remove(vec.begin(),vec.end(),value),vec.end()); //value是要刪除的值
vector<int> v; int i; for( i=0;i<10;i++) v.push_back(i); remove(v.begin(),v.end(),5); remove(v.begin(),v.end(),6); for(vector<int>::iterator it=v.begin();it!=v.end();it++) { cout<<*it<<ends; }
輸出:
0123478999
給一個應用列子:
有 n 個小朋友做游戲,他們的編號分別是 1,2,3...n。他們按照編號從小到大依次順時針圍成一個圓圈,從第一個小朋友開始從 1報數,依次按照順時針方向報數(加一),報 m 的人會離開隊伍,然后下一個小朋友會繼續從 1 開始報數,直到只剩一個小朋友為止。
輸入格式
第一行輸入倆個整數,n,m。(1≤n,m≤1000)
輸出格式
輸出最后一個小朋友的編號,占一行。
樣例輸入
10 5
樣例輸出
3

#include<bits/stdc++.h> using namespace std; vector<int>a; int main() { int n,m; cin>>n>>m; for(int i=1;i<=n;i++) a.push_back(i); int now=1; vector<int>::iterator it; while(a.size()>1) { for(it=a.begin();it!=a.end();) { if(now==m) { it=a.erase(it); now=1; } else { it++; now++; } } } cout<<a[0]<<endl; return 0; }