问题
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 \(O(1)\) 额外空间的条件下完成。
示例 1:
给定 \(nums = [1,1,1,2,2,3]\),
函数应返回新长度 \(length = 5\), 并且原数组的前五个元素被修改为 \(1, 1, 2, 2, 3\) 。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 \(nums = [0,0,1,1,1,1,2,3,3]\),
函数应返回新长度 \(length = 7\), 并且原数组的前五个元素被修改为 \(0, 0, 1, 1, 2, 3, 3\) 。
你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝 int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
解法
使用快慢指针,快指针每次循环都往前推进,慢指针只有在指针位置小于2(前面两位必然需要向前推进)或者慢指针前推2位的数小于快指针的数时,才向前推进,并且将快指针的值赋值到慢指针位置。
源码
class Solution {
public int removeDuplicates(int[] nums) {
int fast=0;
int slow=0;
while(fast<nums.length) {
//慢指针只有在小于2或者前数2位小于快指针时才往前推进
if (slow < 2 || nums[fast] > nums[slow-2]) {
nums[slow] = nums[fast];
slow++;
}
fast++;
}
return slow;
}
}