27. 移除元素

题目

题解

必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成

所以只能修改原数组。

题解一：

直接对满足 nums[i] == val 的值进行依次覆盖即可。

public int removeElement(int[] nums, int val) {

    int idx = 0;
    for (int i = 0; i < nums.length; i++) {
        if (nums[i] != val) {
            nums[idx] = nums[i];
            idx++;
        }
    }

    return idx;
}

题解二：

元素的顺序可以改变

因为对顺序没有要求，所以可以直接使用数组尾部的值覆盖前面的值。

class Solution {
    public int removeElement(int[] nums, int val) {

        int idx = 0;
        int last = nums.length;

        while (idx < last) {
            if (nums[idx] == val) {
                nums[idx] = nums[last - 1];
                last--;
            } else {
                idx++;
            }
        }

        return idx;
    }
}