代码随想录 数组：移除元素

题目：

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

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

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

思路：

1.暴力解法

利用两个for循环，其中一个for循环用于寻找val，另一个循环用于将val后面的元素进行向前覆盖移位。

2.双指针解法（相当于实现了.erase()函数的功能）

一个快指针一个慢指针，快指针用于寻找val的位置，慢指针对应数组的索引，如果快指针找到val，则慢指针不更新，相当于跳过了这个元素。

Leecode 27.移除元素

双指针解法代码如下：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slow = 0;
        for(int fast = 0;fast<nums.size();fast++)
        {
            if(nums[fast]!=val)
            {
                nums[slow] = nums[fast];
                slow++;
            }
        }
        return slow;
    }
};

运行时间为4ms，可以优化一下。

将slow++的操作放入上一行。
 

                nums[slow++] = nums[fast];

暴力解法代码如下：

// 时间复杂度：O(n^2)
// 空间复杂度：O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int size = nums.size();
        for (int i = 0; i < size; i++) {
            if (nums[i] == val) { // 发现需要移除的元素，就将数组集体向前移动一位
                for (int j = i + 1; j < size; j++) {
                    nums[j - 1] = nums[j];
                }
                i--; // 因为下标i以后的数值都向前移动了一位，所以i也向前移动一位
                size--; // 此时数组的大小-1
            }
        }
        return size;

    }
};