[ LeetCode ] 题刷刷(Python)-第27题：移除元素

题目描述

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

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

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

示例

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。
示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,3,0,4]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

解题

解法一： 双指针法

思路

这个题和第26题很像，原理都差不多。

首先初始化 slow 指针为 0，然后使用 for 循环遍历数组（由 fast 指针控制）。在循环内部，判断当前 fast 指针所指向的元素是否等于 val。如果不等于，说明该元素需要保留在新数组中，于是将其赋值给 slow 指针所指向的位置，并将 slow 指针向前推进一位。循环结束后，返回 slow 指针的值作为新数组的长度。

算法复杂度

时间复杂度：O(n)，其中n为nums的长度。

---

空间复杂度：O(1)。

代码

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        # 用于记录已处理过的、不包含 val 的元素的位置
        slow = 0

        for fast in range(len(nums)):
            # 若当前元素不等于 val，则将该元素移动到slow所指向的位置
            if nums[fast] != val:
                nums[slow] = nums[fast]
                slow += 1
        return slow