题目
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
注意:不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定 nums = [1,1,1,2,2,3],
函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。
你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
注意点
1、因为原数组为有序的,所以我们只需要让数组最多出现连续的两个相同的元素;(即不超过两个)
2、因为必须在原数组上进行操作,所以我们可以使用双指针,一个指针(i)指向当前遍历的节点,一个指针(index)指向结果数组的尾节点;
3、因为结果指针(index)一定小于等于当前指针(i),所以操作结果数组不会影响到当前元素的判断。
实现
public int removeDuplicates(int[] nums) {
int index = 0; //结果数组的最后一个下标
for (int i = 1; i < nums.length; i ++){
//如果结果数组中不止一个元素,最后一个元素与当前元素相同,且倒数两个元素相同
if (index > 0 && nums[index] == nums[i] && nums[index - 1] == nums[index]){
//删除当前节点(跳过表示删除)
continue;
}else {
//否则,将当前节点添加到结果数组(替换下一节点表示增加)
nums[++ index] = nums[i];
}
}
//返回结果数组长度
return index + 1;
}