题目:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
使用双指针法
方法一:
执行用时: 36 ms
内存消耗: 13.5 MB
当nums[i] 与给定的值相等时,递增 i 以跳过该元素。只要valnums[i] != val,我们就复制 nums[i] 到 nums[j] 并同时递增两个索引。重复这一过程,直到 i 到达数组的末尾,该数组的新长度为 j。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
i, j = 0, 0
while i < len(nums):
if nums[i] != val:
nums[j] = nums[i]
j += 1
i += 1
return j
方法二:
执行用时: 36 ms
内存消耗: 13.5 MB
当我们遇到 nums[i] = valnums[i]=val 时,我们可以将当前元素与最后一个元素进行交换,并释放最后一个元素。这实际上使数组的大小减少了 1。
请注意,被交换的最后一个元素可能是您想要移除的值。但是不要担心,在下一次迭代中,我们仍然会检查这个元素。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
i, j = 0, len(nums)
while i < j:
if nums[i] == val:
nums[i], nums[j-1] = nums[j-1], nums[i]
j -= 1
else:
i += 1
return j