算法原理
使用双指针法(快慢指针法),可以节省时间复杂度。通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。
定义快慢指针
- 快指针:寻找新数组的元素 ,新数组就是不含有目标元素的数组
- 慢指针:指向更新 新数组下标的位置
两个指针都指向一个数组,fast的任务是找到符合要求的新数组元素,slow则对应 新数组元素的下标。
代码调试过程
列表名是nums,调用元素时要用nums[slow],中括号!!!
修改后的代码:
原因:return slow在while循环里面,当第一次while循环时,3=3,所以slow=0,return回空值。由于return slow在while循环里面,return一次以后就打断了循环。
当在 while 循环内部执行 `return
` 语句时,会立即中断循环并返回相应的值。
所以,`return slow
`语句应该位于循环的外部,以便在遍历完整个列表后返回结果。
继续修改:
原因:fast指针+1的位置不对。如果 `nums[fast]
` 不等于指定的值 `val
`,会执行 `nums[slow] = nums[fast]
`、`slow += 1
` 和 `fast += 1
`,这会导致 `slow
` 和 `fast
` 同时递增。这样会导致处理元素的跳过,因为下一个元素将被放置到 `nums[slow]
` 的位置,但 `slow
` 的值已经递增了。这将导致结果列表中的元素数量少于预期,或者在处理元素时出现错误。
-
`
slow += 1
` 在 `if nums[fast] != val
` 的条件下递增。这是因为只有当 `nums[fast]
` 不等于指定的值 `val
` 时,才需要将该元素移动到 `nums
` 的 `slow
` 索引位置。因此,只有在这个条件下,我们才会递增 `slow
` 的值,以确保下一个非指定元素能够被正确地放置。 -
`
fast += 1
` 则是在每次迭代中都递增。它的目的是在遍历列表时逐个检查元素,并将 `fast
` 递增以处理下一个元素。无论当前元素是否等于指定的值 `val
`,都需要递增 `fast
`,以便在下一次循环中检查下一个元素。
正确的代码
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
slow=0
fast=0
size=len(nums)
while fast<size:
if nums[fast]!=val:
nums[slow]=nums[fast]
slow+=1
fast+=1
return slow
其他
其实这个代码也可以直接调用remove、erase等内嵌函数,但是这样只要一句话就能实现了,显然不是面试所需。
什么时候可以用内嵌函数?
当该函数只是解题的一个步骤,并且我们知道该函数的时间、空间复杂度。