C++标准库std::list按下标删除元素算法实现

1、场景

假设有一个数组[5,8,9,2,4,6],现在我想删除下标为1,3,4的元素,即把元素[8,2,4]删除。

要求使用C++ 标准模板库 list实现(删除元素list比vector效率高)。

2、分析

经查阅资料,发现C++ 模板类list删除元素只有一个erase方法。

iterator erase (iterator position);

iterator erase (iterator first, iterator last);

这个erase函数的形参是iterator类型,迭代器,指向要删除的元素。

下标为1,3,4的元素对应的迭代器值分别为list.begin()+1, list.begin()+3, list.begin()+4,

那么我们是否可以直接调用list.erase(list.begin()+1)、list.erase(list.begin()+3)、list.erase(list.begin()+4)就可以把下标为1,3,4的元素删了呢?答案是不行的!

当我们执行完list.erase(list.begin()+1)后,list的大小已经减一了,这时候的list.begin()+3已经不指向原来的元素了,所以这种做法是不对的。

3、正确代码

#include <vector>
#include <list>
#include <iostream>

using namespace std;

bool sort_desc(int a, int b)
{
return a > b;
}

int main(int argc, char *argv[])
{

    list<int> mylist;
    mylist.push_back(5);
    mylist.push_back(8);
    mylist.push_back(9);
    mylist.push_back(2);
    mylist.push_back(4);
    mylist.push_back(6);

    list<int> index;
    index.push_back(1);
    index.push_back(3);
    index.push_back(4);
    index.sort(sort_desc);//对下标数组进行降序

    for(list<int>::iterator iter=index.begin(); iter!=index.end(); iter++)
    {
        list<int>::iterator delIter=mylist.begin();
        advance(delIter, *iter);//按照由大到小的下标顺序进行删除元素,这样被删除元素的下标是永远不会变的
        mylist.erase(delIter);
    }

    for(list<int>::iterator iter=mylist.begin(); iter!=mylist.end(); iter++)
    {
        cout << *iter << " ";
    }
    cout << endl;

    return 0;
}

运行结果:

5 9 6 

扫描二维码关注公众号,回复: 9202695 查看本文章
发布了200 篇原创文章 · 获赞 37 · 访问量 23万+

猜你喜欢

转载自blog.csdn.net/wo198711203217/article/details/102824450