LeetCode——27（移除元素）

题目链接

移除元素

题目描述

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

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

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

用户评测：

评测机将使用以下代码测试您的解决方案：

int[] nums = [...]; // 输入数组
int val = ...; // 要移除的值
int[] expectedNums = [...]; // 长度正确的预期答案。
                            // 它以不等于 val 的值排序。

int k = removeElement(nums, val); // 调用你的实现

assert k == expectedNums.length;
sort(nums, 0, k); // 排序 nums 的前 k 个元素
for (int i = 0; i < actualLength; i++) {
    assert nums[i] == expectedNums[i];
}

如果所有的断言都通过，你的解决方案将会 通过。

示例 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 个元素之外留下了什么并不重要（因此它们并不计入评测）。

题解思路1：暴力破解

思路：

直接暴力破解，通过两层for循环，第一层先遍历数组找到和目标值相等的元素，之后通过第二层for循环实现将匹配值之后的元素向前移的操作。

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;

    }
};

时间复杂度：O(n^2)
空间复杂度：O(1)

注：做题时我们要记得数组移除并不是物理删除，而是逻辑删除。我们在这里通过后面元素“向前覆盖”的方式实现了移除的效果。

题解思路2：双指针

思路：

通过一个快指针和慢指针实现刚刚两层for循环实现的功能。

快指针负责从头到尾遍历数组，当快指针扫描的数组元素值与目标值不相等时，快慢指针同时实现+1操作，而当快指针扫描的元素等于目标值时，则慢指针不动，只对快指针实现+1操作。

对于比较极端的情况，也是符合的，如数组中只用最后一个元素和目标值匹配，那前面快慢指针是同增的，直到扫描到最后一个元素，此时s慢指针不再+1，快指针+1跳出循环，返回的慢指针数值就是此时新数组大小。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slow = 0; // slow为慢指针
        int length = nums.size();
        for (int fast = 0; fast < length; fast++) {  // fast为快指针
            if (val != nums[fast]) { //快指针所指的元素值不等于目标元素
                nums[slow++] = nums[fast]; 
            }
        }
        return slow;
    }
};

时间复杂度：O(n)
空间复杂度：O(1)