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 nums = [1,1,1,2,2,3],

Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn‘t matter what you leave beyond the new length.

public class Solution {
    public int removeDuplicates(int[] nums) {
        
        int res=0;
        int len=nums.length;
        int set=0;
        
        for(int i=0; i<len; i++){
            if(i==0){
                set++;
                res++;
            }
            else{
                if(nums[i]==nums[i-1]){
                    set++;
                    if(set<=2){
                        res++;
                    }
                }
                else{
                    set=1;
                    res++;
                }
            }

        }
        return res;
    }
}