移除数组中指定元素(Java)

题目描述

给你一个数组 nums 和一个值 val，你需要移除所有数值等于 val 的元素，并返回移除后数组的新长度。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。
示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,3,0,4]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

代码实现

方法一

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Test3 {
    public static void main(String[] args) {
        // 原始数组
        int[] arr = {0, 1, 2, 2, 3, 0, 4, 2};
        int val = 2; // 需要移除的元素值
        int num = removeE(arr, val); // 移除指定元素后的数组长度
        int[] newArr = Arrays.copyOf(arr, num); // 根据移除后的数组长度创建新数组
        List<Integer> list = new ArrayList<>();
        for (int value : newArr) {
            list.add(value);
        }
        System.out.println(num + ", nums = " + list); // 输出移除指定元素后的数组长度和数组内容
    }

    // 移除指定元素的方法
    private static int removeE(int[] arr, int val) {
        int front = 0; // 前指针
        int rear = arr.length; // 后指针
        while (front < rear) {
            if (arr[front] == val) { // 如果当前元素等于要移除的元素值
                arr[front] = arr[rear - 1]; // 将当前元素替换为数组末尾的元素
                rear--; // 数组长度减一
            } else {
                front++; // 继续检查下一个元素
            }
        }
        return front; // 返回移除指定元素后的数组长度
    }
}

思路

使用了双指针的思想，移除数组中指定元素的值。

首先，在 main 方法中，初始化一个整型数组 arr 和一个整数 val，其中 arr 数组包含需要被删除的元素。然后调用 removeE 方法，将 arr 数组和 val 值传递给该方法，以获取更新后的数组长度。接下来，使用 Arrays.copyOf 方法创建一个新数组，该数组将原数组中的前 num 个元素复制到新数组中，其中 num 就是通过 removeE 方法计算出来的需要保留的元素的数量。

在 removeE 方法中，初始化两个指针 front 和 rear 分别指向数组的头和尾。然后我们对数组进行遍历，如果当前元素等于要移除的元素值，就将当前元素替换为数组末尾的元素，然后尾指针 rear  左移一位。如果当前元素不需要被移除，则继续检查下一个元素。当头指针 front 和尾指针 rear 重合的时候，指针遍历完数组中所有的元素。

最后，我们返回另外一个指针 front。这个指针的位置就是最终数组的长度。在 main 方法中，我们使用 Arrays.copyOf 和 List.add 方法创建一个新数组和列表，以存储保留的元素，并输出更新后的数组长度和内容。

总体来说，这段代码使用了简单而有效的双指针算法，时间复杂度为 O(n)，空间复杂度也很小，只需要一个额外的列表就可以了。

方法二

public class Main {
    public static void main(String[] args) {
        int[] nums1 = {3, 2, 2, 3};
        int val1 = 3;
        
        int len1 = removeElement(nums1, val1);
        System.out.println(len1); // 输出为 2
        for (int i = 0; i < len1; i++) {
            System.out.print(nums1[i] + " "); // 输出为 2 2
        }
        System.out.println();

        int[] nums2 = {0, 1, 2, 2, 3, 0, 4, 2};
        int val2 = 2;
        int len2 = removeElement(nums2, val2);
        System.out.println(len2); // 输出为 5
        for (int i = 0; i < len2; i++) {
            System.out.print(nums2[i] + " "); // 输出为 0 1 3 0 4
        }
        System.out.println();
    }
    
    
    public static int removeElement(int[] nums, int val) {
        int i = 0; // 慢指针
        
        // 使用快指针遍历数组
        for (int j = 0; j < nums.length; j++) {
            // 如果当前元素不等于val，则将其移动到慢指针所在位置，并更新慢指针
            if (nums[j] != val) {
                nums[i] = nums[j];
                i++;
            }
        }
        
        return i;
    }
}

思路

removeElement 方法用于移除数组中所有值为 val 的元素，并返回新数组的长度。我们使用两个指针，一个是慢指针 i，一个是快指针 j。快指针遍历数组，如果当前元素不等于 val，它一定是输出数组的一个元素，我们就将快指针 j   指向的元素复制到慢指针 i 位置，然后将两指针同时右移；如果快指针 j 指向的元素等于 val，它不能在输出数组里，此时慢指针不动，快指针右移一位。最后返回慢指针的值作为新的数组长度。