我的菜逼解法:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
for(int i = 0;i < nums.size();i++){
if(nums[i] == val){
nums.erase(nums.begin() + i);
i--;
}
}
return nums.size();
}
};
双指针:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
if(nums.size() == 0) return 0;
int i = 0,j = 0;
while(j < nums.size()){
if(nums[j] == val){
j++;
}
else{
nums[i] = nums[j];
j++;
i++;
}
}
return i;
}
};