题目
给你一个数组
nums
和一个值val
,你需要 原地 移除所有数值等于val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定
nums = [3,2,2,3]
,val = 3
,
函数应该返回新的长度2
, 并且nums
中的前两个元素均为2
。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定
nums = [0,1,2,2,3,0,4,2]
,val = 2
,
函数应该返回新的长度5
, 并且nums
中的前五个元素为0, 1, 3, 0, 4
。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
思路
- 分别定义两个指针点
i
与j
分别指向数组的头和尾 - 接着看
nums[i]
对应的值是否等于val
:
- 如果不是,向后移动.
- 如果是,那么看
nums[j]
是否等于val
:- 如果等于,则j向前移动
- 如果不等于,那么交换
i
与j
的值
i
移动到j
后一位的时候结束循环,此时有效的数组长度即为i
代码
/**
* @param {number[]} nums
* @param {number} val
* @return {number}
*/
var removeElement = function(nums, val) {
if(nums.length === 0) return 0
let i = 0
let j = nums.length - 1
while(i <= j) {
if(nums[i] === val) {
if(nums[j] !== val) {
nums[i] = nums[j] + nums[i]
nums[j] = nums[i] - nums[j]
nums[i] = nums[i] - nums[j]
i++
} else {
j--
}
} else {
i++
}
}
return i
};
复杂度分析
- 时间复杂度 O(N)
假设数组长度为 N,那么执行次数将会是 N+1 次 - 空间复杂度 O(1)