一、题目描述
给你一个数组 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
。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
二、解题思路
- 使用双指针技术。一个指针用于遍历数组(我们称之为
i
),另一个指针用于标记新数组的末尾(我们称之为j
)。 - 遍历数组
nums
,检查每个元素是否等于val
。 - 如果当前元素不等于
val
,将其移动到j
指针指向的位置,然后递增j
。 - 如果当前元素等于
val
,直接递增i
,不做任何操作。 - 遍历完成后,
j
的值即为新数组的长度。
三、具体代码
class Solution {
public int removeElement(int[] nums, int val) {
if (nums == null || nums.length == 0) {
return 0;
}
int j = 0; // 新数组的末尾位置
for (int i = 0; i < nums.length; i++) {
if (nums[i] != val) {
nums[j] = nums[i];
j++;
}
}
return j; // 返回新数组的长度
}
}
四、时间复杂度和空间复杂度
1. 时间复杂度
- 时间复杂度:O(n)。
- 该算法遍历了整个数组一次,对于数组中的每个元素,我们进行了常数时间的操作(比较和可能的赋值)。
- 因此,时间复杂度是 O(n),其中 n 是数组 `nums` 的长度。
2. 空间复杂度
- 空间复杂度:O(1)。
- 算法没有使用额外的数组来存储数据,所有的操作都是在原数组上进行的。
- 我们只使用了两个额外的变量 `i` 和 `j`,它们的空间需求是常数的,不依赖于输入数组的大小。
- 因此,空间复杂度是 O(1),即常数空间复杂度。
五、总结知识点
-
数组处理:代码处理了一个整型数组
nums
,并在数组上进行了原地修改。 -
双指针技术:这是一种常见的算法技巧,用于在遍历数组的同时进行修改。在这里,
i
指针用于遍历数组,而j
指针用于标记新数组的末尾位置。 -
原地操作:代码没有使用额外的数组来存储结果,而是直接在原数组上进行修改,这符合题目要求的 O(1) 额外空间复杂度。
-
条件判断:在遍历过程中,使用
if
语句来判断当前元素是否等于要移除的值val
。 -
数组元素赋值:当当前元素不等于
val
时,将其赋值给j
指针指向的位置,并将j
递增。 -
边界检查:在进行数组操作之前,代码首先检查数组是否为空或长度为0,这是一个良好的编程习惯,可以避免潜在的空指针异常。
-
空值检查:在处理数组之前,代码检查了数组是否为
null
,这是处理可能的空指针异常的一种方式。
以上就是解决这个问题的详细步骤,希望能够为各位提供启发和帮助。