LeetCode 27 Remove Element

LeetCode 27 Remove Element

#include <vector>
using namespace std;
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
int index = 0;
for (int i = 0; i < size; i++){
if (nums[i] != val){
nums[index++] = nums[i];//先取index再++
}
}
for (int i = index; i < size; i++)
nums.erase(nums.begin() + index, nums.end());

return nums.size();
}
};