小言_互联网的博客

vector 中元素的删除

609人阅读  评论(0)

vector 中删除元素的方法是:b.erase(it);  //b是vector, it 是 vector::iterator

但是删除vector 中的元素有些诡异(^_^),稍不注意,就会出错。 

下面先给出基本vector ,如下代码,然后我们来删除为3的元素。

#include <iostream>
#include <vector>
#include <algorithm>
#include <functional>
 
using namespace std;

int main()
{
	int a[7] = {1,2,3,3,4,5,3};
	//通过数组a的地址初始化,注意地址是从0到5(左闭右开区间)
	vector<int> b(a, a+7);
	vector<int>::iterator  it;

	for(it = b.begin(); it != b.end(); it++)
	{
		cout << *it << "\t" ;
	}
	cout << "\n" << endl ;
}

 

第一个代码如下,这个是错误的:

	for(it=b.begin(); it!=b.end(); it++)
	{
		  if( *it == 3)
				 b.erase(it);
	}

乍一看这段代码,很正常。其实这里面隐藏着一个很严重的错误:当b.erase(it)之后,it就变成了一个野指针,对一个野指针进行 it++ 是肯定会出错的。 运行时报异常。

查看MSDN,对于erase的返回值是这样描述的:An iterator that designates the first element remaining beyond any elements removed, or a pointer to the end of the vector if no such element exists,于是改代码:

	for( it=b.begin(); it!=b.end(); it++)
	{
	if( *it== 3)
	it= b.erase(it);
	}

这段代码也是错误的:1)无法删除两个连续的"3"; 2)当3位于vector最后位置的时候,也会出错(在veci.end()上执行 ++ 操作)

正确的代码应该为:

	for( it=b.begin(); it!=b.end();)
	{
	if( *it== 3)
		it= b.erase(it);
	else 
		it++;
	}

在后面添加

for(it = b.begin(); it != b.end(); it++)
	{
		cout << *it << "\t" ;
	}
	cout << "\n" << endl ;

核实我们实现了预期。 1 2 4  5

下面用另一个代码方式实现,这也是正确的代码:

	for(int i=0;i<b.size();)
	{
		if(b[i]==3)
			b.erase(b.begin()+i);
		else
			i++;
	}

 这2个代码的关键是去掉it++ 或者 i++, 这个在代码循环里不删除元素时才运行,都在else 里面。

我最开始是学了 https://blog.csdn.net/daofengdeba/article/details/7865229, 他是我的老师。


转载:https://blog.csdn.net/leon_zeng0/article/details/101321198
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场