【LeetCode面试经典150题】27. 移除元素

一、题目

• 27. 移除元素 - 力扣（LeetCode）
• 给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
• 不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
• 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

二、思路

1. 一眼原地调整数组题，这种题一般都要有循环，用以把后面的符合条件的元素替换当前元素；
2. 那就开始怼循环，首先肯定要遍历数组，拿每个元素与 val 值比对，即遍历操作需要维护一个慢指针；
3. 那交换元素怎么实现呢？应该需要维护一个快指针，慢指针用于固定当前等于 val 的元素位置，快指针用于交换最近的合要求的元素；
4. 那终止条件是什么？快指针触底一次即可，由此得到解法一；
5. 这种方法是两指针都从左到右，那能不能从两边同时向中间逼近呢？
6. 于是设想：慢指针在头部，快指针在尾部，每轮循环开始时快指针指向的值若为 val 则与慢指针交换，慢指针走一步，若不为 val 则快指针走一步，最后结果就是慢指针往左全部是保留的，快指针往后全部是 val ，此时返回 slow 即可，见方法二；

三、解法

解法一

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

解析：首先明确两指针的用途，快指针用于遍历数组查找非 val 值，慢指针用于提示其左侧所有元素都应保留，当快指针找到非 val 值后交换两指针所指元素，慢指针往后走一位；

解法二

class Solution {
    public int removeElement(int[] nums, int val) {
        int len = nums.length;
        int head = 0;
        int tail = len - 1;
        while (head <= tail) {
            if (nums[tail] != val) {
                int tmp = nums[head];
                nums[head++] = nums[tail];
                // 这个地方tail一定不能动！
                // 如果nums[head]不是val的话还要再比较一次
                nums[tail] = tmp;
            } else {
                tail--;
            }
        }
        return head;
    }
}

解析：这里有个要注意的点，为什么返回 head 而不是返回 head + 1 ，这个 head 不是指向数组下标吗？确实是指向数组下标，但是循环终止条件为 head <= tail ，这意味着最后一轮循环时 head 与 tail 会指向同一个元素，并且该元素一定不是 val (这里仔细想想为什么)，于是 head 最后又被加了一次，故这里直接返回 head 即可。

提示：倒数第二轮的情形一定是 slow 指向最后一个非 val 值，而 fast 一定指向最后一个 val 值，所以这轮循环一定进入 else 块，执行 fast--