[LeetCode] Remove Duplicates from Sorted Array II [27]

题目

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].

原题链接（点我）

解题思路

移除数组中反复次数超过2次以上出现的数，可是能够同意反复2次。
这个题类似Remove Duplicates from Sorted Array，第一个想法非常直接就是计数，超过2次的就忽略，根据这个思路的代码见代码一；
上面的思路可行。可是代码看着比較冗余，推断比較多。再来想想原来的数组，该数组是排好序的。假设一个数出现3次以上，那么必有A[i] == A[i-2]。所以依据这个关系能够写出比較精简的代码二。

详见代码。

代码实现

代码一

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(A==NULL || n<=0) return 0;
        int start=1, count = 1, back = A[0];
        for(int i=1; i<n; ++i){
            if(count<2){
                A[start++] = A[i];
                if(back != A[i]){
                    back = A[i];
                    count = 1;
                }else{
                    ++count;
                }
            }else{
                if(A[i] != back){
                    count = 1;
                    A[start++] = A[i];
                    back = A[i];
                }else{
                    ++count;
                }
            }
        }
        return start;
    }
};

代码二

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