leetcode之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
        if (A==NULL || n<=0) {
            return 0;
        }
        int index=0;
        int pre=1;
        while (index<n-1&&pre<=n-1) {
            if (A[index]==A[pre]) {
                pre++;
            } else {
                if (index+1!=pre){
                    A[++index]=A[pre];
                    pre++;
                }
                else {
                    index++;
                    pre++;
                }
            }
        }
        return index+1;
    }
};