序言:今天是第四题啦,Practice makes perfect!加油,每天坚持一下下,胜利就在前方。
删除有序数组中的重复项2-进阶版
给你一个有序数组 nums
,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝 int len = removeDuplicates(nums); // 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。 for (int i = 0; i < len; i++) { print(nums[i]); } 示例 1:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5,并且原数组的前五个元素被修改为1,1,2,2,3。不需要考虑超出新长度后面的元素。
示例 2:
输入:nums = [0,0,1,1,1,1,2,3,3]
输出:7, nums = [0,0,1,1,2,3,3]
解释:函数应返回新长度 length = 7, 并且原数组的前七个元素被修改为 0, 0, 1, 1, 2, 3, 3
。不需要考虑数组中超出新长度后面的元素。
实现思路:我们借助双指针来解决这个问题,一个指针用于遍历,一个指针用于记录数组长度。与昨天的删除有序数组中的重复项有异曲同工之妙。
实现步骤:
-
检查数组长度:注意题目给的是有序数组,如果数组长度小于等于2,直接返回数组的长度,因为所有元素出现次数都不超过两次。
-
初始化指针:使用两个指针
i
和j
。i
用于遍历数组,j
用于记录新数组的长度。 -
处理前两个元素:将前两个不同的元素复制到
j
的位置,因为有序数组至少有两个不同的元素可以保留。 -
遍历数组:从索引 2 开始遍历数组
nums
。 -
更新新数组:
- 如果当前元素
nums[j]
与nums[i]
相同,说明当前元素出现次数超过两次,需要删除多余的元素。 - 如果当前元素
nums[j]
与nums[i]
不同,更新nums[j]
为nums[i]
并递增j
。
- 如果当前元素
-
返回新长度:遍历完成后,
j
的值即为删除重复元素后数组的新长度。
tips:画图演练这个过程有助于理解,如果有人需要的话,可以评论区留言,下次这个体系配上演示图。
实现代码:
public int removeDuplicates(int[] nums) {
if (nums == null || nums.length <= 2) {
return nums.length;
}
int j = 2; // 新数组的长度,初始化为2,保留前两个元素
for (int i = 2; i < nums.length; i++) {
if (nums[i] != nums[j - 2]) {
nums[j++] = nums[i]; // 复制新元素到j的位置
}
}
return j; // 返回新数组的长度
}
这个方法的时间复杂度是 O(n),其中 n 是数组 nums
的长度。空间复杂度是 O(1),因为我们只使用了有限的额外空间(只有 i
和 j
两个指针)。这个方法满足题目要求的“原地”修改数组的条件。
关于题目中的说明,返回的整数是新数组的长度,而输出的答案是数组本身。这是因为在函数内部修改了输入数组,而数组是通过引用传递的,所以对数组的修改对调用者是可见的。调用者可以根据返回的长度打印数组中的元素。