[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)
return n;
int valid = 0;
int dup = 0;
for(int i = 1; i < n; ++ i)
if(A[i] != A[valid]){
dup = 0;
A[++valid] = A[i];
}else{
if(dup <= 0){
A[++valid] = A[i];
++ dup;
}
}
return valid + 1;
}
};