题目大意
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
解题思路
这道题思路和 26题-删除重复项 几乎相同,都用的是快慢指针的方法,区别主要在于判断条件以及指针起始位置。
代码
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
# 快慢指针
if not nums:
return 0
# 排序
nums.sort()
slow, fast = 0, 0
# fast越界即停止
while fast < len(nums):
if nums[fast] != val:
# 将不重复的值按顺序赋值
nums[slow] = nums[fast]
slow += 1
fast += 1
return slow