LeetCode 3208. 交替组 II

3208. 交替组 II

【分析】这个是3101. 交替子数组计数的升级版

这里环可以这样处理，最后一个元素只需要看和前面 k - 1个元素的关系即可

为什么是k - 1呢，如果多了的话，比如k个，那这段序列就包括0 - k了，这样就重复计算了

所以把前面的k - 1个元素复制到后面去

这里需要注意是子数组，中间不能断，一旦有重复元素就要重新计数，另外长度必须超过k才行

class Solution {
    public int numberOfAlternatingGroups(int[] colors, int k) {
        int n = colors.length, m = n + k - 1;
        int[] cols = new int[m];
        for (int i = 0; i < n; i++) cols[i] = colors[i];
        for (int i = 0; i < k - 1; i++) cols[i + n] = colors[i];
        int pre = cols[0];
        int res = 0, t = 1;
        for (int i = 1; i < m; i++) {
            if (pre != cols[i]) {
                t += 1;
                if (t >= k) res += 1;
            } else {
                t = 1;
            }
            pre = cols[i];
        }
        return res;
    }
}