【每日一题】移除元素

🌟个人博客：www.hellocode.top🌟
⭐所有文章均在上方博客首发，其他平台同步更新
🔥本文专栏：《每日一题》
⚡如有问题，欢迎指正，一起学习~~
文章部分参考《代码随想录》，如有侵权，请联系删除~~

---

问题描述

• 时间：2022-05-09
• 题目序号：27
• 难度：简单

问题描述

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

说明

为什么返回数值是整数，但输出的答案是数组呢?

请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说，不对实参作任何拷贝
int len = removeElement(nums, val);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

来源：力扣（LeetCode）

示例1

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

示例2

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

提示

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

解题思路

题解部分参考自代码随想录。如有侵权，请联系进行删除~~

• 在解题时，我们都可以先思考暴力解法进行解题，然后再通过逐步优化，来提高代码的性能
• 对于本题，可以通过暴力和双指针两种方法进行解决

暴力解法

• 本题使用暴力解法，就是需要一个双重for循环，通过对nums数组中所有值进行判断
• 如果等于val值，就通过移动元素将其覆盖（将val后的元素前移），并对size–

*注意：*在将与val值相等位置的元素覆盖后，需要将i–（因为后面元素都向前移动了一位）

双指针解法

• 双指针在数组和链表中的操作还是比较多的，刚开始不易理解，大家可以配合上图多琢磨琢磨
• 定义两个指针（fast和slow）来完成双重for循环完成的任务
• 首先将两个指针都指向第一个元素，然后进行++遍历数组中的元素
• 当fast指向的元素等于val时，slow留在当前位置进行标记，fast进行++开始操作val对应索引后面的元素
• fast指向的元素不等于val时，将fast指向的元素赋给slow所在位置，实现双重for循环中数组前移的操作
• 最后slow的值就是数组在移除元素后的size值

代码实现

通过上述分析，分别可以通过暴力解法和双指针解法两种方式来实现移除元素

暴力解法

class Solution {
    public int removeElement(int[] nums, int val) {
        int size = nums.length;
        for(int i = 0; i < size; i++){
            if(val == nums[i]){
                for(int j = i + 1; j < size; j++){
                    nums[i] = nums[j];
                }
                i--;
                size--;
            }
        }
        return size;
    }
}

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

双指针解法

class Solution {
    public int removeElement(int[] nums, int val) {
        int slow = 0;
        for(int fast = 0; fast < nums.length; fast++){
            if(val != nums[fast]){
                nums[slow++] = nums[fast];
            }
        }
        return slow;
    }
}

• 时间复杂度：O(n)

• 空间复杂度：O(1)

总结

• 一般遇到一个问题时，可以先考虑暴力解法，然后再考虑其他更加高效的算法或者对暴力解法进行不断优化，来提高算法的效率
• 双层for循环遍历数组时，如果在内层循环中对数组元素进行了前后移操作，对应的外层循环控制变量就需要进行相应的调整（–或++）
• 双指针方法在数组和链表中使用的场景还是比较多的，刚开始不太好理解，可以通过动图或者网上查找相应的视频进行理解