题目描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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]);
}
-
双指针法:
- 首先定义一个变量
size
记录新长度,初始left
和right
指针指向0
位置,right
指针不断向右移动直到到达最后一个位置size-1
- 当
nums[left] == nums[right] && right - left + 1 > 2
时,说明这个元素个数超过了2
,那么就将right + 1到size-1
的元素前移一个位置,size--
,right
不用动,因为它的位置被新的元素填充了(继续做判断);如果nums[left] == nums[right] && right - left + 1 <= 2
,right++
;如果nums[left] != nums[right]
,就是遍历到了新的数值,left
跟上,right=left
class Solution {
public int removeDuplicates(int[] nums) {
int size = nums.length;
int left = 0, right = 0;
while(right <= size - 1){
if(nums[left] == nums[right] && right - left + 1 > 2){
moveArray(nums, right + 1, size - 1);
size--;
}else if(nums[left] == nums[right] && right - left + 1 <= 2){
right++;
}else{
left = right;
}
}
return size;
}
//将index从start到end之间的元素前移动一个位置
public void moveArray(int[] nums, int start, int end){
for(int i = start;i <= end;i++){
nums[i - 1] = nums[i];
}
}
}
移动这么多,很耗时
- 简洁省时的双指针:略感抽象
class Solution {
public int removeDuplicates(int[] nums) {
int i = 0;
for (int n : nums)
if (i < 2 || n > nums[i-2])
nums[i++] = n;
return i;
}
}
这种解法,n
是一直往后移动的,而i
会停在不符合条件的元素的位置,等待被重新赋值,再往后走。
- 一开始,
i
和n
是一起走的,直到i
所指向的元素个数超过2
,由于条件if (i < 2 || n > nums[i-2])
,n
开始领先i
,直到n
找到一个大于前面元素的新值,直到找到新值i
会被n
领先i所指向元素的个数 - 1
个位置
例如,1,1,1,1,2,2,3
,i
会停在第三个1
那,直到n
遍历到2
。