题解 | #草原上的牛群分布# 数组模拟

知识点

模拟 计数

思路

有序数组要求相同值不能超过三个，我们可以维护一个cnt记录当前数字有几个，然后从左到右遍历数组，当不满足三个时加入结果数组。由于结果数组一定不大于原数组，可以在原数组上修改。

时间复杂度

遍历一遍数组

AC code (c++)

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int remove_duplicates_v3(vector<int>& nums) {
        int sz = 0;
        int cnt = 0, cur = -1;
        for (auto x : nums) {
            if (cur == x) cnt ++;
            else cur = x, cnt = 1;
            if (cnt <= 3) {
                nums[sz ++] = x;
            }
        }
        nums.resize(sz);
        return sz;
    }
};