LeetCode之移除元素

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例1：给定 nums = [3,2,2,3], val = 3,函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。

示例2：给定 nums = [0,1,2,2,3,0,4,2], val = 2,函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

源代码：

#include<iostream>
#include<string>
#include<vector>
#include<algorithm>

using namespace std;


int removeElement(vector<int>& nums, int val)
{
	for (vector<int>::iterator it = nums.begin(); it != nums.end();)
	{
		if (*it == val)//使用 * 访问迭代器所指向的元素，其实it就是指针
		{
			it = nums.erase(it); //删除指定位置上的元素，也就是删除地址
		}
		else
		{
			++it;
		}
	}
	return nums.size();
}

int main()
{
	vector<int> A = { 3, 2, 2, 2,3 };
	int local=removeElement(A, 3);
	cout << local << endl;	
	system("pause");
	return 0;
}

2
2
2
移除后数组的长度为：3