【Leetcode】【Medium】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].

解题思路1：

遍历A，i表示当前遍历到的数组下标，length表示当前已重新填充的数组下标；设置一个flag用于记录是否已经出现重复；

从i=length=1开始（数组首元素一定不用移除），当A[i] = A[i-1]，如果flag为true，则说明A[i-1]=A[i-2]，因此A[i]需要移除（不对数组做更新操作）；

　　　　　　　　　　　　　　　　　　　　　　  当A[i] = A[i-1]，如果flag为false，则说明A[i-1]!=A[i-2]，因此在当前length位置处使A[length] = A[i]，length++，flag不变；

　　　　　　　　　　　　　　　　　　　　　　  当A[i] != A[i-1]，A[length] = A[i]，length++，flag = false；

最终length指向新数组最后一个元素的下一个位置，因此数值刚好等于新数组个数，直接返回；

解题思路2：

由于代码中有大量的if else 判断语句，非常难看，因此突然想到，直接判断A[i] 是否等于 A[i-2]就可以了，这样可以直接判断出是否连续三个及以上重复，如果重复了3个以上，就不添加到新数组队列中；

但是，由于从i=length=2开始，判断if (A[i] == A[i-2])，A[i-2]有可能已经是新数组的元素，即有可能已经更新的元素值，原数组被破坏了，就无法正确判断是否连续3个；

对于这种向前判断无法成功的例子，往往可以尝试向后判断，即从i=length=0开始，判断if (A[i] == A[i+2])；

至此，问题很轻松的解决了，而且代码行数非常短，由于判断if (A[i] == A[i+2]) 相当于已经判断了最后三个元素的有效性，因此最后的两个元素一定是合法的，不用移除；

代码：

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if (n <= 2) 
            return n;
        int length = 0;
        
        for (int i = 0; i < n - 2; ++i) {
            if (A[i] != A[i+2])
                A[length++] = A[i];
        }
        
        A[length++] = A[n - 2];
        A[length++] = A[n - 1];
        return length;
    }
};