给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
方法一:暴力求解
第一个循环用来遍历,第二个循环用来将排在后面的数组前移
之前尝试了两层for循环但是因为数组的size需要一直变化就没有成功,后来发现for循环遍历过程中i的值并不会因为循环内的操作而改变。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
nums_size = len(nums)
i = 0
while i < nums_size:
if nums[i] == val:
for j in range(i+1,nums_size):
nums[j-1]=nums[j]
nums_size -= 1
i -= 1
i += 1
return nums_size
方法二:双指针
双指针即使用两个指针配合,可以实现节省一层循环的效果。多数教程的说法是一个快指针用来判断一个慢指针用来写入,其实叫读与写更形象一些。直接看比较容易懵不知道为什么快指针会快一步,看代码可知巧妙的地方其实在于循环的判断条件,用一个不等号将与val相等的数值不做处理,快指针按顺序一步一步走将每一个与val不等的数值不断的向前写入慢指针所指向的位置,慢指针每次接受了写入的值之后再向前走这样就形成了一个不包含val的新数组。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
n = len(nums)
low = 0
for fas in range(n):
if nums[fas] != val:
nums[low] = nums[fas]
low += 1
return low