【每日一道算法题】数组移除元素——双指针法

目录

每日一题

1.移除元素

2.数组“删除”的核心

3.双指针法

4.我的刷题实录

4.1.提交第一版

4.2.提交第二版

4.3.提交第三版

4.4.参考答案

每日一题

1.移除元素

力扣题目链接

给你一个数组 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.数组“删除”的核心

关于数组删除元素的理论核心：数组一旦创建，其元素的个数（数组的长度）就不能再发生变化。数组的元素在内存地址中是连续的，不能单独删除数组中的某个元素，只能覆盖。

3.双指针法

什么是双指针法？

双指针，指的是在遍历的过程中，不是普通的使用单个指针进行访问，而是使用两个相同方向（快慢指针）或者相反方向（对撞指针）的指针进行扫描，从而达到相应的目的。

快慢指针也是双指针，但是两个指针从同一侧开始遍历数组，两个指针以不同的策略移动，直到两个指针的值相等（或其他特殊条件）为止。通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。

举个栗子：假设有一个数组[2,3]，要删除的元素是2，删除之后返回新数组的长度。

快慢指针法的过程是怎样的？

快指针做的是常规的遍历操作，有两种情况：

• 当快指针遍历到的值不等于要删除的目标值val，说明该元素可取，将快指针指向的元素赋值给慢指针指向的元素；
• 当快指针遍历到的值等于要删除的值val，此时慢指针不动，快指针继续移动到下一个，然后重复刚才的判断，当遍历的值不等于要删除的目标值val，说明该元素可取，将快指针指向的元素赋值给慢指针指向的元素，然后快指针继续移动，重复刚才的判断。
• 总结：无论如何快指针都会正常遍历，而慢指针是遇到了目标元素就会停止，直到快指针遇到了第一个非目标元素为止，慢指针才会恢复移动。

摘自B站评论区，一种更好理解的方式：但其实slow指针指的并不是新数组的末尾，slow-1才真正指向新数组的末尾。分析过程参考下面“我的刷题实录”。

4.我的刷题实录

4.1.提交第一版

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

        int slow = 0;
        //双指针法，快指针用于正常遍历查询，慢指针用于指向新数组的末尾元素
        for(int fast = 0; fast < nums.length; fast++){
            //当快指针遍历到的值不等于目标值val，那么执行两个操作：
            //1:慢指针将这个值纳入新数组的末尾（赋值））
            //2:指向下一个元素，继续表示新数组的末尾元素（自增）； 
            if(nums[fast] != val){
                nums[slow] == nums[fast];
                slow++;
            }
            //当快指针遍历到的值等于目标值val，那么慢指针停止移动（不能纳入新数组）；
        }

    }
};

显示编译出错，原因是把=赋值不小心写成了==。而且缺少return语句（忘记了题目的要求是返回新数组的长度）。

4.2.提交第二版

class Solution {
    public int removeElement(int[] nums, int val) {
        int slow = 0;
        //双指针法，快指针用于正常遍历查询，慢指针用于指向新数组的末尾元素
        for(int fast = 0; fast < nums.length; fast++){
            //当快指针遍历到的值不等于目标值val，那么执行两个操作：
            //1:慢指针将这个值纳入新数组的末尾（赋值））
            //2:指向下一个元素，继续表示新数组的末尾元素（自增）； 
            if(nums[fast] != val){
                nums[slow] = nums[fast];
                slow++;
            }
            //当快指针遍历到的值等于目标值val，那么慢指针停止移动（不能纳入新数组）；
        }
        //在快指针遍历完成后，直接返回slow的值。sslow表示的是新数组末尾的元素，因此新数组长度为slow+1
        return slow+1;
    }
}

显示解答出错，测试用例显示：

分析了一下原因，关键在于slow的含义没搞清楚。slow实际上并不指向新数组的末尾，想象一下如果快指针此时遍历到了数组的最后一个元素，而且这个元素不是要删除的值，那么就会照常执行赋值、slow自增操作，注意在slow自增前的索引，也就是slow-1这个索引，才指向的是新数组的末尾，而slow一定指向的是新数组末尾后面一个元素    。

还是用上面的图。假设有一个数组[2,3]，要删除的元素是2，删除之后返回新数组的长度。

根据图示来看，slow实际上是在赋值之后，才真正指向新数组的末尾索引，在这个过程之后还会自增，自增后实际上新数组末尾的索引应该是slow-1。而新数组的长度就是slow。

4.3.提交第三版

class Solution {
    public int removeElement(int[] nums, int val) {
        int slow = 0;
        //双指针法，快指针用于正常遍历查询，慢指针用于指向新数组的末尾元素
        for(int fast = 0; fast < nums.length; fast++){
            //当快指针遍历到的值不等于目标值val，那么执行两个操作：
            //1:慢指针将这个值纳入新数组的末尾（赋值）
            //2:指向下一个元素，继续表示新数组的末尾元素（自增）； 
            if(nums[fast] != val){
                nums[slow] = nums[fast];
                slow++;
            }
            //当快指针遍历到的值等于目标值val，那么慢指针停止移动（不能纳入新数组）；
        }
        //在快指针遍历完成后，直接返回slow的值。
        return slow;
    }
}

4.4.参考答案

class Solution {
    public int removeElement(int[] nums, int val) {
        // 快慢指针
        int slowIndex = 0;
        for (int fastIndex = 0; fastIndex < nums.length; fastIndex++) {
            if (nums[fastIndex] != val) {
                nums[slowIndex] = nums[fastIndex];
                slowIndex++;
            }
        }
        return slowIndex;
    }
}

参考的学习资料： 代码随想录