27. 移除元素

题目来源：https://leetcode-cn.com/problems/remove-element

题目

给你一个数组 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. 移除元素》问题的主要内容。