问题描述
给定一个整数数组 nums
和一个整数 val
,要求将数组中所有等于 val
的元素删除,并返回剩余不等于 val
的元素个数。删除元素后,数组前 k 个位置的元素即为新数组内容,不要求保持原有顺序。
快慢指针的核心思想
- 快指针(fast):用于遍历整个数组,每次检查当前元素是否为目标值。
- 慢指针(slow):用于记录下一个存放非目标值的位置。当遇到非目标值时,将其放到慢指针指向的位置,并将慢指针右移一位。
通过这种方式,最终慢指针的索引就代表了新数组的长度,同时数组的前 slow 个位置存放的即是所有不等于 val
的元素。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
slow , fast = 0, 0
while fast < len(nums):
if (nums[fast] == val):
fast += 1
else:
nums[slow] = nums[fast]
fast += 1
slow += 1
return slow