[Leetcode] Remove Duplicates from Sorted Array II

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].

 

Solution:

public class Solution {
    public int removeDuplicates(int[] A) {
        if(A.length<=2)
            return A.length;
        int cnt=1;
        int N=A.length;
        for(int i=2;i<N;++i){
            if(A[i]==A[cnt]&&A[i]==A[cnt-1]){
                
            }else{
                cnt++;
                A[cnt]=A[i];
            }
        }
        return ++cnt;
    }
}

 

posted @ 2015-01-26 04:54  Phoebe815  阅读(141)  评论(0编辑  收藏  举报