数组/移除元素|算法笔记系列02

43 阅读3分钟

移除元素|TypeScript 代码笔记

题目还原

自己回忆:

给定一个nums数组(可能有重复元素)和val。要求删除nums中值为val的元素,并返回新数组的长度。

LeetCode:

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

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

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

思路分析

给出两种方法:暴力解法和双指针法。

第一种:暴力解法

大体思路

  • 第一层for循环:定位到要删除的目标元素。
  • 第二层for循环:从目标元素的下一个元素开始进行向前的覆盖操作。

注意点

  • 每定位到一次目标元素时,就要更新数组的长度。第一层for循环和第二层for循环的边界都依赖最新数组的长度。
  • 每定位到一次目标元素时,第一层for循环的迭代表达式会使得i++。这将会漏掉一个元素。所以需要在每定位到一次目标元素时,主动进行i--操作。

代码解答

const removeElement1 = (nums: number[], val: number): number => {
  let len = nums.length; // 记录最新数组长度
  for (let i = 0; i < len; i++) {
    if (nums[i] === val) {
      // 向前的覆盖操作
      for (let j = i + 1; j < len; j++) {
        nums[j - 1] = nums[j];
      }
      len--; // 更新最新数组长度
      i--; // 防止漏掉一个元素没有遍历
    }
  }
  return len;
};

第二种:双指针法

大体思路

给出一个定义,新数组:删除完所有值为val的元素后的最终数组。

  • 快指针:指向新数组中的元素。
  • 慢指针:指向新数组中的元素要去的位置。

优先遍历快指针,所以for循环的迭代对象是快指针。因为旧数组中的每一个元素都有可能成为新数组中的元素,所以快指针一定要遍历旧数组中的每一个元素。因此for循环的边界是固定的nums.length,区别于暴力解法中的两个for循环的边界是依赖最新数组的长度。

注意点

最初慢指针是0,每次遇到新数组中的元素时,慢指针会加一,当最后一次遇到新数组中的元素时,慢指针又会加一;所以慢指针最终会指在新数组最后一个元素的屁股后面。我们知道数组元素的角标是从0length - 1,所以屁股后面对应的角标刚好是length。所以慢指针最终就是新数组的length

代码解答

const removeElement2 = (nums: number[], val: number): number => {
    let slow = 0, fast = 0;
    for (; fast < nums.length; fast++) {
        if (nums[fast] !== val) {
            nums[slow] = nums[fast];
            slow++;
        }
    }
    return slow;
}

两个代码均运行成功✅,运行结果:

image.png