题目
给你一个整数数组 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 <= 1e5
0 <= colors[i] <= 1
3 <= k <= colors.length
思路
线段树维护区间存在相邻的个数，查询时需要注意，对于边界情况（如果线段树中存在相领记录），需要额外处理
AC代码
struct Tree {
tree: Vec<i32>
}
impl Tree {
pub fn new(len: usize) -> Self {
Tree {
tree: vec![0; len << 2]
}
}
pub fn update(&mut self, node: usize, idx: usize, l: usize, r: usize) {
if l == r {
self.tree[node] = 1;
return;
}
let mid: usize = l + r >> 1;
let l_node: usize = (node << 1) + 1;
let r_node: usize = l_node + 1;
if mid >= idx {
self.update(l_node, idx, l, mid);
} else {
self.update(r_node, idx, mid + 1, r);
}
self.tree[node] = self.tree[l_node] + self.tree[r_node];
}
pub fn query(&mut self, node: usize, q_l: usize, q_r: usize, l: usize, r: usize) -> i32 {
if l > q_r || r < q_l {
return 0;
}
if l >= q_l && r <= q_r {
return self.tree[node];
}
let mid: usize = l + r >> 1;
let l_node: usize = (node << 1) + 1;
let r_node: usize = l_node + 1;
self.query(l_node, q_l, q_r, l, mid) + self.query(r_node, q_l, q_r, mid + 1, r)
}
}
impl Solution {
pub fn number_of_alternating_groups(colors: Vec<i32>, k: i32) -> i32 {
let len: usize = colors.len();
let v_len: usize = len + k as usize;
let mut res: i32 = 0;
let mut v: Vec<i32> = vec![0; v_len];
let mut tree: Tree = Tree::new(v_len << 2);
let mut hash: Vec<bool> = vec![false; len];
for i in 0 .. len {
v[i] = colors[i];
}
for i in 0 .. (k as usize) {
v[len + i] = v[i];
}
for i in 1 .. v_len {
if v[i] == v[i - 1] {
tree.update(0, i - 1, 0, v_len - 1);
tree.update(0, i, 0, v_len - 1);
}
}
for i in 0 ..= len {
if hash[i % len] {
continue;
}
let q_all: i32 = tree.query(0, i, i + k as usize - 1, 0, v_len - 1);
let q_l: i32 = tree.query(0, i, i, 0, v_len - 1);
let q_r: i32 = tree.query(0, i + k as usize - 1, i + k as usize - 1, 0, v_len - 1);
if q_all == 0 || q_all == q_l || q_all == q_r || q_all == q_l + q_r {
res += 1;
hash[i % len] = true;
}
}
res
}
}
推荐几款学习编程的免费平台
免费在线开发平台（docs.ltpp.vip/LTPP/）
探索编程世界的新天地，为学生和开发者精心打造的编程平台，现已盛大开启！这个平台汇集了近4000道精心设计的编程题目，覆盖了C、C++、JavaScript、TypeScript、Go、Rust、PHP、Java、Ruby、Python3以及C#等众多编程语言，为您的编程学习之旅提供了一个全面而丰富的实践环境。
在这里，您不仅可以查看自己的代码记录，还能轻松地在云端保存和运行代码，让编程变得更加便捷。平台还提供了私聊和群聊功能，让您可以与同行们无障碍交流，分享文件，共同进步。不仅如此，您还可以通过阅读文章、参与问答板块和在线商店，进一步拓展您的知识边界。
为了提升您的编程技能，平台还设有每日一题、精选题单以及激动人心的编程竞赛，这些都是备考编程考试的绝佳资源。更令人兴奋的是，您还可以自定义系统UI，选择视频或图片作为背景，打造一个完全个性化的编码环境，让您的编程之旅既有趣又充满挑战。
免费公益服务器（docs.ltpp.vip/LTPP-SHARE/…）
作为开发者或学生，您是否经常因为搭建和维护编程环境而感到头疼？现在，您不必再为此烦恼，因为一款全新的免费公共服务器已经为您解决了所有问题。这款服务器内置了多种编程语言的编程环境，并且配备了功能强大的在线版VS Code，让您可以随时随地在线编写代码，无需进行任何复杂的配置。
随时随地，云端编码
无论您身在何处，只要有网络连接，就可以通过浏览器访问这款公共服务器，开始您的编程之旅。这种云端编码的便利性，让您的学习或开发工作不再受限于特定的设备或环境。
丰富的编程语言支持
服务器支持包括C、C++、JavaScript、TypeScript、Go、Rust、PHP、Java、Ruby、Python3以及C#等在内的多种主流编程语言，满足不同开发者和学生的需求。无论您是初学者还是资深开发者，都能找到适合自己的编程环境。
在线版VS Code，高效开发
内置的在线版VS Code提供了与本地VS Code相似的编辑体验，包括代码高亮、智能提示、代码调试等功能，让您即使在云端也能享受到高效的开发体验。
数据隐私和安全提醒
虽然服务器是免费的，但为了保护您的数据隐私和安全，我们建议您不要上传任何敏感或重要的数据。这款服务器更适合用于学习和实验，而非存储重要信息。
免费公益MYSQL（docs.ltpp.vip/LTPP-SHARE/…）
作为一名开发者或学生，数据库环境的搭建和维护往往是一个复杂且耗时的过程。但不用担心，现在有一款免费的MySQL服务器，专为解决您的烦恼而设计，让数据库的使用变得简单而高效。
性能卓越，满足需求
虽然它是免费的，但性能绝不打折。服务器提供了稳定且高效的数据库服务，能够满足大多数开发和学习场景的需求。
在线phpMyAdmin，管理更便捷
内置的在线phpMyAdmin管理面板，提供了一个直观且功能强大的用户界面，让您可以轻松地查看、编辑和管理数据库。
数据隐私提醒，安全第一
正如您所知，这是一项公共资源，因此我们强烈建议不要上传任何敏感或重要的数据。请将此服务器仅用于学习和实验目的，以确保您的数据安全。
免费在线WEB代码编辑器（docs.ltpp.vip/LTPP-WEB-ID…）
无论你是开发者还是学生，编程环境的搭建和管理可能会占用你宝贵的时间和精力。现在，有一款强大的免费在线代码编辑器，支持多种编程语言，让您可以随时随地编写和运行代码，提升编程效率，专注于创意和开发。
多语言支持，无缝切换
这款在线代码编辑器支持包括C、C++、JavaScript、TypeScript、Go、Rust、PHP、Java、Ruby、Python3以及C#在内的多种编程语言，无论您的项目需要哪种语言，都能在这里找到支持。
在线运行，快速定位问题
您可以在编写代码的同时，即时运行并查看结果，快速定位并解决问题，提高开发效率。
代码高亮与智能提示
编辑器提供代码高亮和智能提示功能，帮助您更快地编写代码，减少错误，提升编码质量。
免费二维码生成器（docs.ltpp.vip/LTPP-QRCODE…）
二维码（QR Code）是一种二维条码，能够存储更多信息，并且可以通过智能手机等设备快速扫描识别。它广泛应用于各种场景，如：
企业宣传
企业可以通过二维码分享公司网站、产品信息、服务介绍等。
活动推广
活动组织者可以创建二维码，参与者扫描后可以直接访问活动详情、报名链接或获取电子门票。
个人信息分享
个人可以生成包含联系方式、社交媒体链接、个人简历等信息的二维码。
电子商务
商家使用二维码进行商品追踪、促销活动、在线支付等。
教育
教师可以创建二维码，学生扫描后可以直接访问学习资料或在线课程。
交通出行
二维码用于公共交通的票务系统，乘客扫描二维码即可进出站或支付车费。 功能强大的二维码生成器通常具备用户界面友好，操作简单，即使是初学者也能快速上手和生成的二维码可以在各种设备和操作系统上扫描识别的特点。