【编程题】——移除元素

题目：给定一个整形数组，删除其中等于val的元素，然后返回删除后的数组

代码：

# include<iostream>
# include<vector>
# include<iterator>
using namespace std;

void removeElement(vector<int>& nums, int val) 
 {
	 vector<int>::iterator it;
	 for(it=nums.begin(); it != nums.end();)
	 {
		 if(*it == val)
		 {
			 it = nums.erase(it);
		 }
		 else
		 {
			 ++it;
		 }
	 }    
 }

int main()
{
	vector<int>arr;
	int tmp[]={1,2,5,4,2,5,6,5,9,1};
	for(int i=0;i<sizeof(tmp)/sizeof(tmp[0]);i++)
	{
		arr.push_back(tmp[i]);
	}
	removeElement(arr,5);
	for(int i=0;i<arr.size();i++)
	{
		cout<<arr[i]<<" ";
	}
	cout<<endl;
	return 0;
}