[LeetCode] Remove Duplicates from Sorted Array II

int removeDuplicates2(int A[], int n) {if (n < 3){return n;}int index = 1, length = n, times = 1,lastnum = A[0];for (int i = 1; i < n; i++){if (A[i] == lastnum){times++;if (times <= 2){A[index] = A[i];index++;}else{length--;}}else{A[index] = A[i];index++;lastnum = A[i];times = 1;}}return length;}