leetcode-27. 移除元素

leetcode-27. 移除元素

题目：

代码：

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

int removeElement(vector<int>& nums, int val) {
	if(val>50){
		return nums.size();
	}
	int p=0,q=0;
	while(q<nums.size()){
		if(nums[q]!=val){
			if(p!=q){
				nums[p]=nums[q];
			}
			p++;
		}
		q++;
	}
	return p;
}

int main(){
	int len;
	vector<int> nums;
	int val;
	string s;
	int tn=0;
	getline(cin,s);
	cin>>val;
	for(int i=0;i<s.size();i++){
		if(s[i]==' '){
			nums.push_back(tn);
			tn=0;
		}else{
			tn=tn*10+s[i]-'0';
		}
	}
	nums.push_back(tn);
	len=removeElement(nums, val);
	cout<<len<<endl;
	for(int i=0;i<len;i++){
		cout<<nums[i]<<" ";
	}
	return 0;
}