题目
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝 int len = removeElement(nums, val); // 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。 for (int i = 0; i < len; i++) { print(nums[i]); }
示例
示例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
。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
提示
class Solution { public int removeElement(int[] nums, int val) { int n = 0; for (int i = 0; i < nums.length; i++) { if (nums[i] != val) { nums[n] = nums[i]; n++; } } return n; }}
详细解读
这个算法是用来从一个整数数组 nums 中移除所有值等于给定整数 val 的元素,并返回新数组的长度。
使用两个指针 i 和 n,其中 i 用于遍历原始数组 nums,而 n 用于跟踪新数组中的位置。
从头开始遍历 nums 数组,对于每个元素 nums[i],进行以下检查:
如果 nums[i] 不等于 val,将其复制到新数组的位置 n,然后递增 n。
如果 nums[i] 等于 val,则跳过,不将其包含在新数组中。
继续遍历整个数组,直到 i 到达数组的末尾。
返回 n 的值,这个值表示新数组中的不包含等于 val 的元素的长度。
这个算法的核心思想是通过遍历数组一次,将不等于 val 的元素逐个复制到新数组的开头,同时保持新数组的长度,以达到删除所有等于 val 的元素的目的。这是一种高效的方式,因为它只需要一次遍历数组,时间复杂度为 O(n),其中 n 是数组的长度。
举个例子,如果原始数组 nums 为 [3, 2, 2, 3, 4, 5, 2],并且 val 为 2,使用这个算法后,新数组会变成 [3, 3, 4, 5],同时返回值 n 为 4,表示新数组的长度。
时间复杂度为O(n)