Orion's Studio.

算法(3)-移除元素与快慢指针

2024/03/06

题目(easy):

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

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

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

思路

暴力解法就是两层遍历,第一层遍历数组,第二层更新数组,时间复杂度是 O(n^2)。

双指针法

通过一个快指针和一个慢指针在一个 for 循环完成两个 for 循环的工作。

1
2
3
4
5
6
7
8
9
const removeElement = (nums, val) => {
let k = 0;
for (let i = 0; i < nums.length; i++) {
if (nums[i] !== val) {
nums[k++] = nums[i];
}
}
return k;
}

时间复杂度:O(n)。
空间复杂度:O(1)。

CATALOG
  1. 1. 题目(easy):
  2. 2. 思路
    1. 2.1. 双指针法