移除元素(三 8.31)

52 阅读1分钟

移除元素(三 | 8.31)

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

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

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

思路:定义两个指针,一个是i,用来遍历数组,一个是index用来填充数组。

public class RemoveElement {
    public static int removeElement(int[] nums, int val) {
        int index = 0; // 指向待填充的位置
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != val) {
                nums[index] = nums[i];
                index++;
            }
        }
        return index; // 返回新数组的长度
    }

    public static void main(String[] args) {
        int[] nums = {3, 2, 2, 3, 4, 5, 3};
        int val = 3;
        int newLength = removeElement(nums, val);

        // 输出移除指定元素后的数组
        for (int i = 0; i < newLength; i++) {
            System.out.print(nums[i] + " ");
        }
    }
}