27. 移除元素
题目
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
解题思路
思路:双指针
审题,本题的要求是【不要使用额外的数组空间】。
定义双指针,front ,tail。当 nums[front] 的值与给定的 val 的值相等时,则 front 指针向前移动,跳过该元素。若是不相等,则将 nums[front] 的值复制给 nums[tail],然后两个指针向前同时移动。
循环重复这个过程,但 front 到达末尾时,返回的新数组长度即为 tail。
具体过程看以下图解:
具体代码如下。
代码实现
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
# 定义双指针
front = 0
tail = 0
# 循环遍历数组
while front < len(nums):
# 当 front 指针所对应的值与给定 val 值不同时
# front 指针向前移动,跳过该元素
if nums[front] != val:
# 将 front 所对应的值复制给 tail 指针所对应的值
nums[tail] = nums[front]
tail += 1
front += 1
return tail
实现结果
以上就是使用双指针的方法,移动指针时,判断元素值是否与给定的值相同,决定是否跳过该元素。若不同则将前指针的值复制给后指针对应的值,否则跳过该元素,重复该过程,从而解决《27. 移除元素》问题的主要内容。