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

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n==0)
            return 0;
        int start=1;
        bool twice=false;
        for(int i=1;i<n;i++){
            if(A[i]!=A[i-1]){
                 A[start++]=A[i];
                 twice=false;
            }
            else if(!twice){
                A[start++]=A[i];
                twice=true;
            }
        }
        return start;
    }
};

 

posted @ 2014-11-06 14:40  li303491  阅读(89)  评论(0编辑  收藏  举报