LeetCode 每日一题 [21] 移除元素

LeetCode 移除元素 [简单]

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-element

示例 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。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。

题目分析

解法1：

要求就地删除给定值的所有元素，可以使用双指针，一个快指针，一个慢指针，当 nums[j] 和 val 相等时候，递增 j 跳过此元素，然后当不想等的时候，就移动 res 并且用不想等的值覆盖前面的值，

解法2：

当要删除的元素很少的时候，现在考虑数组包含很少的要删除的元素的情况。例如，num=[1，2，3，5，4]，Val=4num=[1，2，3，5，4]，Val=4。之前的算法会对前四个元素做不必要的复制操作。另一个例子是 num=[4，1，2，3，5]，Val=4num=[4，1，2，3，5]，Val=4。似乎没有必要将 [1，2，3，5][1，2，3，5] 这几个元素左移一步，因为问题描述中提到元素的顺序可以更改。

代码实现

public class LeetCode_21_RemoveElement {
    public static void main(String[] args) {
        int[] nums = {3, 2, 2, 3};
        int res = removeElement(nums, 2);
        for (int i = 0; i < res; i++) {
            System.out.print(nums[i] + " ");
        }
        System.out.println();
        System.out.println("==================");
        int[] nums2 = {3, 2, 2, 3};
        int res2 = removeElement02(nums2, 2);
        for (int i = 0; i < res2; i++) {
            System.out.print(nums2[i] + " ");
        }
    }

    public static int removeElement02(int[] nums, int val) {
        int res = 0;
        int n = nums.length;
        while (res < n) {
            if (nums[res] == val) {
                nums[res] = nums[n - 1];
                n--;
            } else {
                res++;
            }
        }
        return res;
    }

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