力扣27.移除元素

力扣链接

题目介绍

解题方法

法一：双指针法

先设定变量 idx，指向待插入位置，idx 初始值为 0。然后从题目的「要求/保留逻辑」出发，来决定当遍历到任意元素 i 时，应该做何种决策：
1.如果当前元素 i 与移除元素 val 相同，那么跳过该元素。
2.如果当前元素 i 与移除元素 val 不同，那么我们将其放到下标 idx 的位置，并让 idx 自增右移。
最终得到的 idx 即是答案。

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

法二：利用元素顺序可以改变

前面需要移除的元素找后面的元素将其替换掉

class Solution {
    public int removeElement(int[] nums, int val) {
        int left = 0;
        int right = nums.length-1;
        while (left <= right) {
            if (nums[left] == val) {
                nums[left] = nums[right];
                right--;
            } else {
                left++;
            }
        }
        return left;
    }
}