执行结果:通过
执行用时和内存消耗如下:




代码如下:
int numberOfAlternatingGroups(int* colors, int colorsSize, int k) {
int res = 0, cnt = 1;
for (int i = -k + 2; i < colorsSize; i++) {
if (colors[(i + colorsSize) % colorsSize] != colors[(i - 1 + colorsSize) % colorsSize]) {
cnt += 1;
} else {
cnt = 1;
}
if (cnt >= k) {
res += 1;
}
}
return res;
}
解题思路:
- 初始化变量 :
res用于存储结果,即交替颜色组的数量。cnt用于记录当前考虑的连续交替颜色的长度。
- 循环遍历 :
- 使用一个从
-k + 2到colorsSize - 1的循环变量i。这个范围的选择是为了确保能够检查每个可能的长度为k的子数组,同时避免数组越界。 - 循环的起始值
-k + 2是因为我们需要查看当前元素与其前面k-1个元素的关系,而由于数组是循环的(或者可以看作循环的),所以我们需要从稍微早于数组开始的位置开始考虑。
- 使用一个从
- 处理循环数组 :
- 使用
(i + colorsSize) % colorsSize和(i - 1 + colorsSize) % colorsSize来处理数组索引的循环。这确保了即使i是一个负数或超出了数组的实际长度,我们仍然可以正确地访问数组元素。
- 使用
- 交替颜色计数 :
- 对于每个
i,比较colors[(i + colorsSize) % colorsSize]和colors[(i - 1 + colorsSize) % colorsSize]。 - 如果这两个颜色不相同,说明当前元素延续了交替颜色序列,因此
cnt增加 1。 - 如果这两个颜色相同,说明交替颜色序列被打破,因此重置
cnt为 1。
- 对于每个
- 更新结果 :
- 每次
cnt达到或超过k时,说明找到了一个长度为k或更长的交替颜色组,因此res增加 1。
- 每次
- 返回结果 :
- 循环结束后,返回
res,即交替颜色组的总数。
- 循环结束后,返回
注意:
- 这个方法有一个潜在的问题:它实际上会计算所有长度至少为
k的交替颜色序列的数量,而不仅仅是长度为k的。如果只需要长度为k的交替颜色组的数量,需要在cnt达到k时立即重置它,而不是继续增加cnt并检查它是否超过k。不过,根据函数名numberOfAlternatingGroups和没有特定说明只计算长度为k的组,这里的实现可能是合理的。 - 另外,循环的起始值
-k + 2和结束值colorsSize - 1确保了所有可能的k长度的子数组都被检查到了,并且处理好了数组的循环性。