LeetCode 27.移除元素

题目描述

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

• 更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
• 返回 k。

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2,_,_]
解释：你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3,_,_,_]
解释：你的函数应该返回 k = 5，并且 nums 中的前五个元素为 0,0,1,3,4。
注意这五个元素可以任意顺序返回。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

提示：

• 0 <= nums.length <= 100
• 0 <= nums[i] <= 50
• 0 <= val <= 100

思路

本题首先可以使用暴力解法或者库函数erase()解法，前者的时间复杂度为O(n^2)，前者的时间复杂度为O(n)。而笔者使用的是双指针法，诶嘿，时间复杂度为O(n)。

关键在于定义快慢指针：

• 快指针：寻找新数组的元素 ，新数组就是不含有目标元素的数组
• 慢指针：指向新数组中需要更新的下标的位置

代码

C++版：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slow=0;
        for(int fast=0;fast<nums.size();fast++){
            // slow 用来收集不等于 val 的值，如果 fast 对应值不等于 val，则把它与 slow 替换
            if(nums[fast]!=val){
                nums[slow]=nums[fast];
                slow++;
            }
        }
        // 当快指针指向非目标元素时，不更新新数组
        return slow;
    }
};

Python版：

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        slow=0
        for fast in range(0,len(nums)):
            # slow 用来收集不等于 val 的值，如果 fast 对应值不等于 val，则把它与 slow 替换
            if nums[fast]!=val:
                nums[slow]=nums[fast]
                slow+=1
            # 当快指针指向非目标元素时，不更新新数组
        return slow

需要注意的地方

1.数组的元素是不能删除的，只能覆盖。

2.使用双指针的时候要明确两个指针各自的意思。

3.本题中慢指针最终指向的索引是新数组的长度。