80. Remove Duplicates from Sorted Array II

public class Solution {
    public int removeDuplicates(int[] nums) {
        int cur = 1;
        int count = 1;
        for(int i = 1; i < nums.length; i++){
            if(nums[i] == nums[i - 1]){
                count ++;
                if(count <= 2){
                    nums[cur ++] = nums[i];
                }
            }else{
                nums[cur ++] = nums[i];
                count = 1;
            }
        }
        return cur;
    }
}