题目描述
给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
暴力解
[!NOTE] 重点
- 当找到移除的值，将后续元素依次向前覆盖
代码示例
from typing import List
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
i = 0
size = len(nums)
while i < size:
if nums[i] == val: # 查找到移除值时，将后续元素向前覆盖
j = i + 1
while j < size:
nums[j - 1] = nums[j]
j += 1
size -= 1
i -= 1
i += 1
return size
# 示例用法
if __name__ == "__main__":
solution = Solution()
# 示例 1
nums1 = [3, 2, 2, 3]
val1 = 3
result1 = solution.removeElement(nums1, val1)
print(f"示例 1：移除值 {val1} 后数组 {nums1[:result1]}，新长度为 {result1}")
# 示例 2
nums2 = [0, 1, 2, 2, 3, 0, 4, 2]
val2 = 2
result2 = solution.removeElement(nums2, val2)
print(f"示例 2：移除值 {val2} 后数组 {nums2[:result2]}，新长度为 {result2}")
双指针
[!NOTE] 重点
- 快指针：寻找新数组的元素 ，新数组就是不含有目标元素的数组
- 慢指针：指向更新 新数组下标的位置
代码示例
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
# 初始化慢指针
slow = 0
# 遍历整个数组
for fast in range(len(nums)):
# 当快指针指向的元素不等于目标值时，将其赋值给慢指针位置
if nums[fast] != val:
nums[slow] = nums[fast]
# 慢指针移动一步
slow += 1
# 返回新的长度
return slow
# 示例用法
if __name__ == "__main__":
solution = Solution()
# 示例 1
nums1 = [3, 2, 2, 3]
val1 = 3
len1 = solution.removeElement(nums1, val1)
print(f"新的长度：{len1}, 数组：{nums1[:len1]}")
# 示例 2
nums2 = [0, 1, 2, 2, 3, 0, 4, 2]
val2 = 2
len2 = solution.removeElement(nums2, val2)
print(f"新的长度：{len2}, 数组：{nums2[:len2]}")
