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 查看本文章