vector中erase用法注意事项



vector::erase()：从指定容器删除指定位置的元素或某段范围内的元素
vector::erase()方法有两种重载形式
如下：
iterator erase(   iterator _Where);
iterator erase(   iterator _First,   iterator _Last);
如果是删除指定位置的元素时：
返回值是一个迭代器，指向删除元素下一个元素;
如果是删除某范围内的元素时：返回值也表示一个迭代器，指向最后一个删除元素的下一个元素;

看下面的程序，目的是删除数组里面的所有值为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;
           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;
        itor=array.erase(itor);
    }
    else
    {
        itor++;
    }
}

或者：

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

也可以使用remove方法：
array.earse( remove(array.begin(), array.end(),6),  array.end() );