Leetcode 3208. Alternating Groups II

• Leetcode 3208. Alternating Groups II
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3208. Alternating Groups II

1. 解题思路

这一题同样只需要想明白问题的核心就行了，这道题的核心就在于说找到所有的交替不同的最大字串，假设其长度为n，则能够构成的长度为k的group数目就是$n-k+1$，考虑到可以循环，因此我们只需要在原数组当中加入头部的前$k-1$个元素即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def numberOfAlternatingGroups(self, colors: List[int], k: int) -> int:
        n = len(colors)
        colors = colors + colors[:k-1]
        idx = 0
        ans = 0
        while idx < n+k-2:
            while idx < n+k-2 and colors[idx] == colors[idx+1]:
                idx += 1
            cnt = 1
            while idx < n+k-2 and colors[idx] != colors[idx+1]:
                idx += 1
                cnt += 1
            ans += max(cnt-k+1, 0)
        return ans

提交代码评测得到：耗时700ms，占用内存20.7MB。