前言
力扣第二十七题 移除元素
如下所示:
给你一个数组 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,4,0,3]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
一、思路
这一题其实跟 力扣第二十六题-删除有序数组中的重复项
有一点类似,都是原地修改数组,并返回处理后的数组的长度,可以忽略数组中超出长度后面的元素。
这一题的实现思路就是 遍历整个数组,将所有不相等的元素聚集到数组的左边,因为超出长度的元素无需处理,所以并不需要实际移除相等的元素,采取覆盖的策略即可。
具体的步骤如下所示:
- 记录新数组的下标
current
,current默认值为0。结果ret = 0
- 从数组 nums 的第二个元素开始遍历,判断当前元素与目标元素
- 如相等:不做任何处理。
- 如不相等:如前面有元素删除过,即
i != current
,将nums[current] = nums[i]
。然后current++
,ret++
- 返回
ret
即可
实际例子分析
此处以示例1中的 nums = [3,2,2,3], val = 3
为例。
- 初始值
current = 0
,ret = 0
- 开始遍历,
nums[0] == val (3 == 3)
,所以不做处理 nums[1] != val (2 != 3)
且i != current (1 != 0)
,将nums[0] = 2
。然后current = 1, ret = 1
nums[2] != val (2 != 3)
且i != current (2 != 1)
,将nums[1] = 2
。然后current = 2, ret = 2
- 开始遍历,
nums[3] == val (3 == 3)
,所以不做处理 - 返回结果
ret = 2
(其实此时数组nums = [2, 2, 2, 3]
,但因可以忽略超长度的元素,故无影响)
二、实现
实现代码
public int removeElement(int[] nums, int val) {
int ret = 0;
int current = 0; // 当前位置
for (int i=0; i<nums.length; i++) {
if (val != nums[i]) {
if (i != current) { // 前面有元素删除过
nums[current] = nums[i];
}
ret++;
current++;
}
}
return ret;
}
测试代码
public static void main(String[] args) {
int[] nums = {0,0,1,1,1,2,2,3,3,4};
new Number26().removeDuplicates(nums);
}
结果
三、总结
感谢看到最后,非常荣幸能够帮助到你~♥