vector中erase用法注意事项

最新推荐文章于 2024-04-23 17:15:19 发布

小白的学习笔记

最新推荐文章于 2024-04-23 17:15:19 发布

阅读量911

点赞数

分类专栏： C/C++ 面试/数据结构

C/C++ 同时被 2 个专栏收录

58 篇文章

订阅专栏

面试/数据结构

3 篇文章

订阅专栏

本文详细介绍了C++ STL中vector::erase()方法的两种重载形式及其使用注意事项，通过实例展示了如何正确地使用该方法来删除指定元素，并对比了erase与remove方法的应用场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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;  
           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() );