代码随想录 移除元素

移除元素这个问题确实比较简单，原理也很直接。

问题描述

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

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

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

主要思路

核心的任务就是在不新建一个数组的前提下，使原本数组中符合要求的元素全部排在左边。那么，对应的方式也就有两种：

1. 将符合要求的元素从左边一个一个放置；
2. 或是将不符合要求的全部放到右边。

这里用到经典的方法就是双指针，两个指针分别有不同的作用。

1. 一个指针的作用是遍历整个数组，搜寻满足或是不满足的节点；
2. 另一个指针的作用是保证数组的纯度，它能确保它的左边或是右边的数组是满足某种要求的(全保存或是全舍去)。

那么对应着原本思考的两种方法来说，对应的实现方法便也能对应。

代码实现

方法一：

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;
    }
}

快指针遍历数组，而慢指针的作用是从左边开始，逐个储存满足要求的数组。这里的慢指针保证了该指针左边的所有元素都是满足要求的。

方法二：

class Solution {
    public int removeElement(int[] nums, int val) {
        int left = 0;
        int right = nums.length - 1;
        while(right >= 0 && nums[right] == val) right--; //将right移到从右数第一个值不为val的位置
        while(left <= right) {
            if(nums[left] == val) { //left位置的元素需要移除
                //将right位置的元素移到left（覆盖），right位置移除
                nums[left] = nums[right];
                right--;
            }
            left++;
            while(right >= 0 && nums[right] == val) right--;
        }
        return left;
    }
}

左指针遍历，而右指针将所有不满足要求的元素都放在其右边。