【leetcode】【80】Remove Duplicates from Sorted Array II

一、问题描述

Follow up for "Remove Duplicates":

What if duplicates are allowed at most twice?

For example,

Given sorted array nums = 

[1,1,1,2,2,3]

,

Your function should return length = 

5

, with the first five elements of nums being 

1

, 

1

, 

2

, 

2

 and 

3

.
It doesn't matter what you leave beyond the new length.

二、问题分析

题目要求最终的数组中前边部分满足要求即可，与Remove Duplicates一样，说明还是把后边的满足条件的数字往前挪。鉴于该题要求相同数字不能超过两次，显然如果还和Remove Duplicates一样使用一个指针来遍历很难满足要求，那么我们可以考虑双指针方法。从这个例子[1,1,1,2,2,3]来分析，前两个元素不需要管，从第三个数字来看，我们可以设置两个指针pre，cur，将cur指向第三个元素，将pre指向第二个元素，显然只有当nums[cur]==nums[pre]&&nums[cur]==nums[pre-1]的时候我们需要将cur继续往后移动寻找不同的元素；这时nums[cur]==2，不满足之前的判断条件，那么需要将2覆盖pre之后的元素，所以数组变为了[1,1,2,2,2,3]，这时pre指向第一个2，cur指向第三个2，继续执行判断条件，pre可以++，这时虽然讲第三个2覆盖了第二个2因为是循环随意可能会出现同值覆盖的情况，然后继续执行后面的。有点啰嗦，看代码。

三、Java AC代码

public int removeDuplicates(int[] nums) {
if (nums.length <= 2)
return nums.length;

int prev = 1; // point to previous
int curr = 2; // point to current

while (curr < nums.length) {
if (nums[curr] == nums[prev] && nums[curr] == nums[prev - 1]) {
curr++;
} else {
prev++;
nums[prev] = nums[curr];
curr++;
}
}

return prev + 1;
}