题目描述
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素 最多出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
注意,其中数组以引用的方式传递。
样例
示例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 。 不需要考虑数组中超出新长度后面的元素。
思路
使用双指针的思路。定义两个指针 p1 和 p2。
p1 表示当前需要存放元素的位置(慢指针),p2 表示当前遍历到等待检查的元素的位置(快指针)。
如果 p1 - 2位置的元素与 p2 位置的元素值不同,那么说明 nums[p2] 可以放在 p1位置,如果相同则不放,说明 nums[p2] 这个元素值已经放过2个了。
代码
class Solution {
public int removeDuplicates(int[] nums) {
int p1,p2;
int len = nums.length;
if(len <= 2)
return len;
p1 = p2 = 2;
while(p2 < len) {
if(nums[p1-2] != nums[p2]) {
nums[p1++] = nums[p2];
}
p2++;
}
return p1;
}
}