当前位置：首页 > article >正文

【从零开始的LeetCode-算法】3206. 交替组 I

article 2024/11/28 18:02:26

给你一个整数数组 colors ，它表示一个由红色和蓝色瓷砖组成的环，第 i 块瓷砖的颜色为 colors[i] ：

colors[i] == 0 表示第 i 块瓷砖的颜色是红色。
colors[i] == 1 表示第 i 块瓷砖的颜色是蓝色。

环中连续 3 块瓷砖的颜色如果是交替颜色（也就是说中间瓷砖的颜色与它左边和右边的颜色都不同），那么它被称为一个交替组。

请你返回交替组的数目。

注意，由于 colors 表示一个环，第一块 瓷砖和 最后一块 瓷砖是相邻的。

示例 1：

输入：colors = [1,1,1]

输出：0

解释：

示例 2：

输入：colors = [0,1,0,0,1]

输出：3

解释：

交替组包括：

提示：

3 <= colors.length <= 100
0 <= colors[i] <= 1

我的解答：

class Solution {
    public int numberOfAlternatingGroups(int[] colors) {
        int size = colors.length - 1;
        int res = 0;
        for(int i = 0 ; i <= size; i++){
            // int left = (i + size) % (size + 1);
            // int right = (i + 2) % (size + 1);
            int left = i == 0 ?  size : i - 1;
            int right = i == size ? 0 : i + 1;
            if(colors[left] == colors[right] && colors[left] != colors[i]) res++;
        }
        return res;
    }
}