题目:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
【第一种方法】
思路解析:
数组是用连续的一段内存空间,来存储相同数据类型数据的集合。因为这个“连续内存存储”,使得数组在插入或者删除的元素的时候,其它元素也要相应的移动。因此不能单纯的使用删除元素函数
1.循环访问数组中的元素
2.如果数组元素等于移除元素,弹出该元素
3.保持指针的访问位置不变(使用continue实现)
class Solution(object):
def removeElement(self, nums, val):
"""
:type nums: List[int]
:type val: int
:rtype: int
"""
i = 0
while i < len(nums):
if nums[i] == val:
nums.pop(i)
continue
i+=1
return len(nums)
【第二种方法:快慢指针】
思路解析:
快指针循环数组指向对比元素,慢指针指向被赋值的位置
1.快慢指针同时指向数组的第一个元素
2.当快指针指的元素不等于val,将nums[fast]赋值给nums[slow],并将快慢指针同时向前移动一位
3.当快指针指向的元素等于val,表示该元素为移除元素,将快指针向前移动,慢指针的位置不动
4.最后返回慢指针的大小就是数组的长度(slow指向的是将赋值的位置,所以slow代表数组长度)
class Solution(object):
def removeElement(self, nums, val):
if len(nums) == 0:
return 0
fast = 0
slow = 0
while fast < len(nums):
if nums[fast] != val:
nums[slow] = nums[fast]
fast += 1
slow += 1
else:
fast += 1
return slow