vector::erase():从指定容器删除指定位置的元素或某段范围内的元素

vector::erase()方法有两种重载形式

如下:

iterator erase(   iterator _Where);
iterator erase(   iterator _First,   iterator _Last);

如果是删除指定位置的元素时:

返回值是一个迭代器,指向删除元素下一个元素;

如果是删除某范围内的元素时:返回值也表示一个迭代器,指向最后一个删除元素的下一个元素;

也就是说:

vector在erase之后,指向下一个元素的位置,其实进行erase操作时将后面所有元素都向前移动,迭代器位置没有移动。itor=array.erase(itor) erase返回下一个元素的地址,相当于给itor一个新值。


看下面的程序,目的是删除数组里面的所有值为6的元素:

#include<iostream>  
#include<vector>  
using namespace std;  
int main()  
{  
    vector<int> array;  
    array.push_back(1);  
    array.push_back(6);  
    array.push_back(3);  
    array.push_back(6);  
    array.push_back(6);  
    array.push_back(2);  
  
    vector<int>::iterator itor;  
    vector<int>::iterator itor2;  
    for(itor=array.begin();itor!=array.end();)  
    {  
        if(6==*itor)  
        {  
           itor2=itor;  
           array.erase(itor2);  
  
        }  
        itor++;  
    }  
    itor=array.begin();  
    for(itor=array.begin();itor!=array.end();)  
    {  
        cout<<(*itor++);  
    }  
    getchar();  
    return 0;  
}

运行结果输出1362,可见其中一个6并未删除,这是迭代器的问题。

原因在于erase以后,itor已经指向下一个元素了,不应该在itor++,否则会跳过下一个元素,即连续两个6时跳过了第二个6. 

另外,在itor2=itor时,两个itor是一样的,这样做并无意义。可修改如下:

vector<int>::iterator itor;  
// vector<int>::iterator itor2;  
for(itor=array.begin();itor!=array.end();)  
{  
    if(6==*itor)  
    {  
        // itor2=itor;  
        array.erase(itor);  
    }  
    else  
    {  
        itor++;  
    }  
}

或者:


vector<int>::iterator itor;  
for(itor=array.begin();itor!=array.end();itor++)  
{  
    if(6==*itor)  
    {  
        array.erase(itor);  
        itor--;  
    }  
}


也可以使用remove方法:

array.earse( remove(array.begin(), array.end(),6),  array.end() );