Java

RemoveDuplicatesfromSortedArrayII.java

/*
Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

For example,
Given sorted array A = [1,1,1,2,2,3],

Your function should return length = 5, and A is now [1,1,2,2,3].
*/

public class Solution {
    public int removeDuplicates(int[] A) {
        if (A==null){
            return 0;
        } 
        
        if (A.length<3){
            return A.length;
        }
        
        int temp=A[1];
        int len=1;
        
        for(int i=2; i<A.length; i++){
            if (A[i]!=A[i-2]){
                A[len++]=temp;
                temp=A[i];
            }
        }
        
        A[len++]=temp;
        
        return len;
    }
}