Leetcode 27 移除元素 Remove Element

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-element
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法一：erase函数

找到该元素后直接原地删除，erase函数会修改size，最后直接返回删完的size即可。

注意：这里使用了while语句，如果改成if会跳过某些元素从而WA。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        for(int i=0;i<nums.size();i++){
            while(nums[i]==val && i<nums.size()){
                nums.erase(nums.begin()+i);
            }
        }
        return nums.size();
    }
};

先对数组进行排序，这样可以省略比val大的元素。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        sort(nums.begin(),nums.end());
        int i=0;
        while(i<nums.size()){
            if(nums[i]==val){
                nums.erase(nums.begin()+i);
            }else if(nums[i]>val){
                break;
            }else(i++);
        }
        return nums.size();
    }
};

方法二：双指针

参照上一题的方法，可以通过相似的方法得到如下代码：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int i=0,j=0,size=nums.size();
        while(j<size){
            if(nums[j]==val){
                j++;
            }else{
                nums[i++]=nums[j++];
            }
        }
        return i;
    }
};