LeetCode | Remove Duplicates from Sorted Array II

Follow up for "Remove Duplicates":

What if duplicates are allowed at most twice?

For example,

Given sorted array A = 

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

,

Your function should return length = 

5

, and A is now 

[1,1,2,2,3]

.

//思想类似于前天的去数组重复，用length标记满足重复最多允许两次的数组的长度
public class Solution {
public int removeDuplicates(int[] A) {
if (A.length <= 2) return A.length;

int length = 2;
int prepre = A[0];
int pre = A[1];

for (int i=2; i<A.length; i++){          // for (int i=2; i<A.length; i++){
if (A[i] != prepre){                 //     if (A[i] != A[i-2]){
A[length++] = A[i];              //         A[length++] = A[i];
}                                    //      }
prepre = pre;                        // }
pre = A[i];
}                                        // return length;
return length;
}
}

注解：注意右边的程序是有错误的。

由于题目允许在数组中重复两次，且数组是sorted的，故每次for循环取A[i]与它前面的前面相比较，如果两者相等，则中间的肯定也相等，即为一个不符合要求的三重复。但注意：不能写成右边的 A[i]!=A[i-2]形式。

举例：数组1 1 1 2 2 3

在i=3的循环后，数组变为 1 1 2 2 2 3，length=3，A[2]=2，而prepre是前一次i=3循环时由pre赋给的，prepre=原A[2]=1，这点很重要。

在i=4的循环时，判断A[4]!=prepre，而不是判断A[4]!=A[2]，否则只有一个三重复的数组由于判断A[2]=A[4]而又导入了一个额外的三重复。

prepre与pre是用来在消除重复赋值前保存原数组的值的。