删除排序数组中的重复数字 II

跟进“删除重复数字”：如果可以允许出现两次重复将如何处理？

给出数组A =[1,1,1,2,2,3]，你的函数应该返回长度5，此时A=[1,1,2,2,3]。

public class Solution {
    /**
     * @param A: a array of integers
     * @return : return an integer
     */
    public int removeDuplicates(int[] nums) {
        // write your code here
         if (nums.length == 0){
            return 0;
        }
        int n = 0;
        int count = 0;//用来做一个标记 1表示已经有>=2个出现了
        for(int i=1 ; i < nums.length; i++){
            if(nums[i] == nums[n] && count == 1){
                count += 1;//多于两个重复之后就count+略过即可
            }
            if(nums[i] == nums[n] && count == 0){
                nums[n+1] = nums[i];
                n = n+1;//遇到第一个开始重复的数字 n+1  连续重复n就不再变化
                count = 1;
            }
            
            if(nums[i] != nums[n] ){
                nums[n+1] = nums[i];
                n=n+1;
                count = 0;
            }
        }
       return n+1;
    }
}