【LetMeFly】3206.交替组 I：遍历

力扣题目链接：https://leetcode.cn/problems/alternating-groups-i/

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

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

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

请你返回 交替 组的数目。

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

 

示例 1：

输出：0

解释：

示例 2：

输出：3

解释：

交替组包括：

 

提示：

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

解题方法：遍历

遍历“三连砖”的起点：

假设起点是$i$，那么下一个瓷砖就是$(i+1)\%len(colors)$，下下个瓷砖就是$(i+2)\%len(colors)$.

• 时间复杂度$O(len(colors))$
• 空间复杂度$O(1)$

AC代码

C++

class Solution {
public:
    int numberOfAlternatingGroups(vector<int>& colors) {
        int ans = 0;
        for (int i = 0; i < colors.size(); i++) {
            if (colors[i] != colors[(i + 1) % colors.size()] && colors[i] == colors[(i + 2) % colors.size()]) {
                ans++;
            }
        }
        return ans;
    }
};

Python

from typing import List

class Solution:
    def numberOfAlternatingGroups(self, colors: List[int]) -> int:
        return sum(colors[i] != colors[(i + 1) % len(colors)] != colors[(i + 2) % len(colors)] for i in range(len(colors)))

Java

class Solution {
    public int numberOfAlternatingGroups(int[] colors) {
        int ans = 0;
        for (int i = 0; i < colors.length; i++) {
            if (colors[i] != colors[(i + 1) % colors.length] && colors[i] == colors[(i + 2) % colors.length]) {
                ans++;
            }
        }
        return ans;
    }
}

Go

package main

func numberOfAlternatingGroups(colors []int) (ans int) {
    for i := range colors {
        if colors[i] != colors[(i + 1) % len(colors)] && colors[i] == colors[(i + 2) % len(colors)] {
            ans++
        }
    }
    return
}

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/144071026