力扣刷题：面试经典150题（2）

原题链接：移除元素

题目描述：

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

• 更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
• 返回 k。

测试样例：

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2,_,_]
解释：你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3,_,_,_]
解释：你的函数应该返回 k = 5，并且 nums 中的前五个元素为 0,0,1,3,4。
注意这五个元素可以任意顺序返回。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

解法一：遍历删除

只需要遍历数组，当当前值为val时调用vector的erase方法，即可完成。

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

这里有一点需要注意的是当我们调用erase()方法时会导致所有元素的下标减一，所以我们每次删除之后需要将i减一，调整下标。除此还可采用迭代器写法。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        auto it = nums.begin();
        while (it != nums.end()) {
            if (*it == val) {
                it = nums.erase(it);  // erase 会返回下一个元素的迭代器
            } else {
                ++it;
            }
        }
        return nums.size();  // 返回最后数组的大小，即剩余元素的数量
    }
};

时间复杂度：最好情况下O(n)，最坏情况下O(n²)（当所有元素都需要删除时）

空间复杂度：O(1)

解法二：双指针

由于题目中说“你在返回的 k 个元素之外留下了什么并不重要”，所以我们可以将所有值不为val的元素将其放在数组的头部。所以可以采用快慢双指针来进行操作。

定义两个指针slow和fast，先同时指向数组开始，fast指针向后扫描，当遇到目标元素，slow指针指向的元素替换为fast指针指向的元素，然后前移一步slow指针，直到fast指针指向数组末尾，则代表操作完成。

如：nums = [0,1,2,2,3,0,4,2], val = 2

1. 首先fast指针在0位置，元素0不等于2，将其赋值给slow指针指向的下标0处，然后移动两个指针。
2. fast指针在1处，符合条件，将其赋值给slow指针处，随后移动两个指针。
3. fast指针在2处，元素为2，等于val，则不做操作，慢指针不动，只移动快指针。
4. 移动快指针到4处，元素为3，则将其赋值给慢指针指向的下标2处，操作后数组为[0,1,3,2,3,0,4,2]。以此类推最后便可操作完毕，得到数组[0,1,3,0,4,0,4,2]。

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

时间复杂度：O(n)

空间复杂度：O(1)