27 移除元素
题目27 移除元素
思路:
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组
要知道数组的元素在内存地址中是连续的,不能单独删除数组中的某个元素,只能覆盖。
双指针法(快慢指针法): 通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。
定义快慢指针:
快指针:寻找新数组的元素 ,新数组就是不含有目标元素的数组
慢指针:指向更新 新数组下标的位置
代码
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
slow = 0
fast = 0
nums_len = len(nums)
while fast < nums_len:
if nums[fast] == val:
fast += 1
else:
nums[slow] = nums[fast]
fast += 1
slow += 1
return nums[slow] # 不是nums[slow+1],因为上面循环的时候slow多循环了一步