[Leetcode] Remove Duplicates from Sorted Array II

class Solution {public:    int removeDuplicates(int A[], int n) {        // Start typing your C/C++ solution below        // DO NOT write int main() function        if (n >= 0 && n <= 2) return n;                int front = 0;        int count = 1;        for (int i = 1; i < n; ++i)        {            if (A[i] != A[front])            {                A[++front] = A[i];                count = 1;            }            else            {                if (count == 1)                {                    A[++front] = A[i];                    ++count;                }            }        }                return front + 1;    }};