remove duplicates from sorted array

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