class Solution {
public:
int numberOfAlternatingGroups(vector<int>& colors, int k) {
}
};
100351. 交替组 II
给你一个整数数组 colors
和一个整数 k
,colors
表示一个由红色和蓝色瓷砖组成的环,第 i
块瓷砖的颜色为 colors[i]
:
colors[i] == 0
表示第 i
块瓷砖的颜色是 红色 。colors[i] == 1
表示第 i
块瓷砖的颜色是 蓝色 。环中连续 k
块瓷砖的颜色如果是 交替 颜色(也就是说除了第一块和最后一块瓷砖以外,中间瓷砖的颜色与它 左边 和 右边 的颜色都不同),那么它被称为一个 交替 组。
请你返回 交替 组的数目。
注意 ,由于 colors
表示一个 环 ,第一块 瓷砖和 最后一块 瓷砖是相邻的。
示例 1:
输入:colors = [0,1,0,1,0], k = 3
输出:3
解释:
交替组包括:
示例 2:
输入:colors = [0,1,0,0,1,0,1], k = 6
输出:2
解释:
交替组包括:
示例 3:
输入:colors = [1,1,0,1], k = 4
输出:0
解释:
提示:
3 <= colors.length <= 105
0 <= colors[i] <= 1
3 <= k <= colors.length
原站题解
golang 解法, 执行用时: 85 ms, 内存消耗: 7.5 MB, 提交时间: 2024-07-08 09:31:53
func numberOfAlternatingGroups(colors []int, k int) (ans int) { n := len(colors) cnt := 0 for i := 0; i < n*2; i++ { if i > 0 && colors[i%n] == colors[(i-1)%n] { cnt = 0 } cnt++ if i >= n && cnt >= k { ans++ } } return }
python3 解法, 执行用时: 184 ms, 内存消耗: 20 MB, 提交时间: 2024-07-08 09:31:20
class Solution: def numberOfAlternatingGroups(self, colors: List[int], k: int) -> int: n = len(colors) ans = cnt = 0 for i in range(n * 2): if i > 0 and colors[i % n] == colors[(i - 1) % n]: cnt = 0 cnt += 1 if i >= n and cnt >= k: ans += 1 return ans
java 解法, 执行用时: 6 ms, 内存消耗: 61.8 MB, 提交时间: 2024-07-08 09:30:46
public class Solution { public int numberOfAlternatingGroups(int[] colors, int k) { int n = colors.length; int ans = 0; int cnt = 0; for (int i = 0; i < n * 2; i++) { if (i > 0 && colors[i % n] == colors[(i - 1) % n]) { cnt = 0; } cnt++; if (i >= n && cnt >= k) { ans++; } } return ans; } }
cpp 解法, 执行用时: 77 ms, 内存消耗: 98.6 MB, 提交时间: 2024-07-08 09:30:33
class Solution { public: int numberOfAlternatingGroups(vector<int>& colors, int k) { int n = colors.size(); int ans = 0, cnt = 0; for (int i = 0; i < n * 2; i++) { if (i > 0 && colors[i % n] == colors[(i - 1) % n]) { cnt = 0; } cnt++; ans += i >= n && cnt >= k; } return ans; } };