难度简单
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
LeetCode里面给出的是快慢指针,循环整个字符串。快指针负责将等于val的元素带向数列末端,慢指针负责拷贝快指针交换过来的内容。
只要快指针不等于val, 就将快慢指针对应的列表元素互换,并且慢指针向前走一步;
如果快指针等于val,则慢指针不动,快指针向后走一步;
循环上一步骤。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
fast = 0
slow = 0
lenOfList = len(nums)
while fast < lenOfList:
if nums[fast] != val:
nums[slow] = nums[fast]
fast += 1
slow += 1
else:
fast += 1
return slow
于是翻了下LeetCode里面的题解和评论,大家貌似用的都是快慢指针,只有我用了左右指针……
左右指针的方法来源于二分法,这个方法相对来说有点复杂。发博客来纪念一下自己清奇的脑回路。
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
left = 0
lenOfNum = len(nums)
right = lenOfNum - 1
timesOfVal = 0
mid = lenOfNum // 2
if lenOfNum == 1 and nums[0] == val:
return 0
while left <= right:
if nums[left] == val and nums[right] != val:
nums[left], nums[right] = nums[right], nums[left]
left += 1
right -= 1
timesOfVal += 1
elif nums[left] == val and nums[right] == val:
right -= 1
elif nums[left] != val and nums[right] == val:
right -= 1
left += 1
else:
left += 1
return right + 1
该代码旨在将左侧的val值放在右侧,其实思路蛮好理解:
当左指针指向数字不等于val的时候:有两种情况:1. 右侧指针也不等于val。 2. 右侧指针等于val。
我们什么时候需要互换呢?
右侧既然不等于,那么就先放置不动,左指针继续向前移动,直到碰到val,再与右侧指针互换值。
其他情况可以按照这个思路一一来看。