移除元素

刀刀

3/20/2025

0 字

0 分钟

题目

描述

给一个数组 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,,]

• 示例 2:

  输入: nums = [0,1,2,2,3,0,4,2], val = 2 输出: 5, nums = [0,1,4,0,3,,,_]

链接地址

leedCode链接

答题思路

这里有两个考点，第一个是要修改原数组，第二个是返回不等于val的元素数量。

修改原数组意味着不能使用 filter 这种方法，因为 filter 是返回一个新的数组，而不是修改原数组。这里采取的是双指针的方法，一个指针用来遍历数组，另一个指针用来指向不等于val的元素。 判断第一个指针是否等于val，如果不等于则跳过自增一，如果等于则将第二个指针的值赋给第一个指针，然后第二个指针自减一。最后返回第一个指针的值即可。

解答代码

/**
 * @param {number[]} nums
 * @param {number} val
 * @return {number}
 */
var removeElement = function(nums, val) {
    let left = 0, right = nums.length - 1
    while(left <= right) {
        if (nums[left] === val) {
            nums[left] = nums[right]
            right--
        }
        else {
            left++
        }
    }
    return left
};