【Java 算法实现】移除数组元素（双指针）

问题描述

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

要求：

• 不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并原地修改输入数组。
• 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

示例

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

提示：

• 0 <= nums.length <= 100
• 0 <= nums[i] <= 50
• 0 <= val <= 100

解决方案

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

public class SolutionTest {
    public static void main(String[] args) {
        Solution solution = new Solution();
        
        // 测试案例 1
        int[] nums1 = {3, 2, 2, 3};
        int val1 = 3;
        int newLength1 = solution.removeElement(nums1, val1);
        System.out.println("New Length: " + newLength1); // 应输出 2
        System.out.print("Modified Array: ");
        for (int i = 0; i < newLength1; i++) {
            System.out.print(nums1[i] + " "); // 应输出 2 2
        }
        System.out.println();

        // 测试案例 2
        int[] nums2 = {0, 1, 2, 2, 3, 0, 4, 2};
        int val2 = 2;
        int newLength2 = solution.removeElement(nums2, val2);
        System.out.println("New Length: " + newLength2); // 应输出 5
        System.out.print("Modified Array: ");
        for (int i = 0; i < newLength2; i++) {
            System.out.print(nums2[i] + " "); // 可能的输出 0 1 3 0 4
        }
        System.out.println();

        // 测试案例 3：所有元素都需要移除
        int[] nums3 = {2, 2, 2};
        int val3 = 2;
        int newLength3 = solution.removeElement(nums3, val3);
        System.out.println("New Length: " + newLength3); // 应输出 0
        System.out.print("Modified Array: ");
        for (int i = 0; i < newLength3; i++) {
            System.out.print(nums3[i] + " ");
        }
        System.out.println();

        // 测试案例 4：没有元素需要移除
        int[] nums4 = {1, 2, 3, 4};
        int val4 = 5;
        int newLength4 = solution.removeElement(nums4, val4);
        System.out.println("New Length: " + newLength4); // 应输出 4
        System.out.print("Modified Array: ");
        for (int i = 0; i < newLength4; i++) {
            System.out.print(nums4[i] + " "); // 应输出 1 2 3 4
        }
        System.out.println();

        // 测试案例 5：空数组
        int[] nums5 = {};
        int val5 = 1;
        int newLength5 = solution.removeElement(nums5, val5);
        System.out.println("New Length: " + newLength5); // 应输出 0
        System.out.print("Modified Array: ");
        for (int i = 0; i < newLength5; i++) {
            System.out.print(nums5[i] + " ");
        }
        System.out.println();
    }
}