Leetcode 27：数组——移除元素

1. 题目

leetcode链接
给你一个数组 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。

2. 思路

2.1 暴力解法

这个题目暴力的解法就是两层for循环，一个for循环遍历数组元素 ，第二个for循环更新数组。

• 时间复杂度：$O(n^2)$
• 空间复杂度：$O(1)$

2.2 双指针法

快指针遍历数组，慢指针更新数组

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$

3. 代码实现

3.1 暴力求解

class Solution {
    
    
public:
    int removeElement(vector<int>& nums, int val) {
    
    
        int size = nums.size();

        // 循环遍历数组元素
        for (int i = 0; i < size; i++){
    
    
            if (nums[i] == val){
    
    
                // 循环更新数组元素
                for(int j = i + 1; j < size; j++){
    
    
                    nums[j - 1] = nums[j]; // 注意这里需要更新（删除）的元素从i位置开始
                }
                // 删除一个元素之后，数组的长度-1， (i+1)位置的元素移到了i位置
                i--;
                size--;
            }
        }
        return size;

    }
};

3.2 双指针法

class Solution {
    
    
public:
    int removeElement(vector<int>& nums, int val) {
    
    
        int fastIndex = 0;
        int slowIndex;

        // fastIndex遍历数组元素
        for (slowIndex = 0; fastIndex < nums.size(); fastIndex++){
    
    
            if(nums[fastIndex] != val){
    
    
                // slowIndex更新数组元素
                nums[slowIndex] = nums[fastIndex];
                slowIndex++;
            }
        }

        return slowIndex;
        

    }
};