题解 | #草原上的牛群分布#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param nums int整型一维数组
     * @return int整型
     */
    public int remove_duplicates_v3 (int[] nums) {
        // write code here
        // 双指针法：i指向下一个要存放的位置，j向后遍历
        // i 仅当 j 遍历同一个牛群的前三个时，才移动，并放入j对应的元素
        // j 一直往后移动

        if (nums.length == 0) return 0;
        if (nums.length <= 3) return 3;

        int i = 0, j = 1;
        int count = 1;
        while(j < nums.length){
            if (nums[j] == nums[j - 1]) {
                if (count < 3) {
                    nums[++i] = nums[j];
                    count++;
                }
            } else {
                nums[++i] = nums[j];
                count = 1;
            }
            j++;
        }

        return i + 1;
    }
}