Leetcode JAVA刷刷站（27）移除元素

news2025/1/12 12:29:04

一、题目概述

二、思路方向

为了解决这个问题，你可以使用双指针的方法来原地移除所有等于 val 的元素，并返回不等于 val 的元素数量。

三、代码实现

public class Solution {  
    public int removeElement(int[] nums, int val) {  
        int left = 0; // 左指针，指向不等于val的元素应该存放的位置  
        for (int right = 0; right < nums.length; right++) { // 右指针，遍历数组  
            if (nums[right] != val) { // 如果当前元素不等于val  
                nums[left] = nums[right]; // 将当前元素放到左指针的位置  
                left++; // 左指针向右移动一位  
            }  
        }  
        return left; // 左指针的位置就是不等于val的元素数量  
    }  
  
    public static void main(String[] args) {  
        Solution solution = new Solution();  
        int[] nums = {3, 2, 2, 3};  
        int val = 3;  
        int k = solution.removeElement(nums, val);  
        System.out.println("不等于val的元素数量：" + k);  
        System.out.println("修改后的数组（前k个元素）：");  
        for (int i = 0; i < k; i++) {  
            System.out.print(nums[i] + " ");  
        }  
    }  
}

执行结果：

四、小结

在这个实现中，我们使用两个指针，left 和 right。left 指针始终指向下一个不等于 val 的元素应该存放的位置，而 right 指针用于遍历整个数组。当遍历到不等于 val 的元素时，我们将其复制到 left 指针指向的位置，并将 left 指针向右移动一位。这样，遍历结束后，数组的前 left 个位置就是所有不等于 val 的元素，而 left 的值就是这些元素的总数。

这种方法的时间复杂度是 O(n)，其中 n 是数组 nums 的长度，因为我们只需要遍历数组一次。空间复杂度是 O(1)，因为我们直接在原数组上进行修改，没有使用额外的空间（除了几个用于迭代的变量）。