2026-06-14:切换打开灯泡。用go语言,给定一个整数数组 bulbs,数组中每个元素都在 1 到 100 之间。共有 100 个电灯泡,编号从 …
🤖 AI总结
主题
使用Go语言实现灯泡切换算法,通过哈希表和异或操作判断最终亮灯状态。
摘要
本文介绍了用Go语言解决灯泡切换问题的算法,通过哈希表和异或操作高效判断最终亮灯状态,并给出了多种语言实现。
关键信息
- 1 使用异或操作记录灯泡切换奇偶次数
- 2 通过哈希表存储灯泡状态
- 3 最终对亮灯编号升序排序
2026-06-14:切换打开灯泡。用go语言,给定一个整数数组 bulbs,数组中每个元素都在 1 到 100 之间。共有 100 个电灯泡,编号从 1 到 100,初始时全部处于关闭状态。
依次遍历数组 bulbs 的每个值 bulbs[i],对编号为 bulbs[i] 的灯泡执行切换操作:
• 若该灯泡原本关闭,则将其打开;
• 若该灯泡原本打开,则将其关闭。
遍历完成后,收集所有最终处于打开状态的灯泡编号,并按升序排列输出;如果最终没有任何灯泡打开,则返回空列表。
1 <= bulbs.length <= 100。
1 <= bulbs[i] <= 100。
输入: bulbs = [10,30,20,10]。
输出: [20,30]。
解释:
第 bulbs[0] = 10 个灯泡当前是关闭状态,将其打开。
第 bulbs[1] = 30 个灯泡当前是关闭状态,将其打开。
第 bulbs[2] = 20 个灯泡当前是关闭状态,将其打开。
第 bulbs[3] = 10 个灯泡当前是打开状态,将其关闭。
最终,第 20 个和第 30 个灯泡处于打开状态。
题目来自力扣3842。
一、整体执行分步详细过程 阶段1:遍历灯泡操作数组,用哈希map记录每个灯泡切换次数的奇偶性
核心逻辑:灯泡初始全关,切换偶数次=最终关闭,切换奇数次=最终打开;使用异或^=1快速标记奇偶,1代表切换奇数次(亮),0代表偶数次(灭)。
1. 初始化空字典cnt,用于存储灯泡编号与切换奇偶标记;
2. 读取数组第一个元素10:
• 字典中无键10,cnt[10] = 0 ^ 1 = 1;
• 含义:10号灯泡切换1次,奇数,暂时亮起。
当前字典:{10:1}
3. 读取数组第二个元素30:
• 字典中无键30,cnt[30] = 0 ^ 1 = 1;
• 含义:30号灯泡切换1次,奇数,暂时亮起。
当前字典:{10:1, 30:1}
4. 读取数组第三个元素20:
• 字典中无键20,cnt[20] = 0 ^ 1 = 1;
• 含义:20号灯泡切换1次,奇数,暂时亮起。
当前字典:{10:1, 30:1, 20:1}
5. 读取数组第四个元素10:
• 字典已有键10,对应值为1,cnt[10] = 1 ^ 1 = 0;
• 含义:10号灯泡累计切换2次,偶数,最终熄灭。
遍历全部数组后最终字典:{10:0, 30:1, 20:1}
阶段2:筛选所有最终亮起的灯泡
遍历哈希字典里每一组(灯泡编号,标记值):
1. 键10,值0:标记不大于0,代表切换偶数次,灯泡关闭,舍弃;
2. 键30,值1:标记大于0,切换奇数次,灯泡打开,将30存入结果切片ans;
3. 键20,值1:标记大于0,切换奇数次,灯泡打开,将20存入结果切片ans;
筛选完成后,未排序的ans切片内容:[30,20]
阶段3:对结果升序排序
调用排序函数对ans切片从小到大重排:
原切片[30,20]排序后变为[20,30]
阶段4:返回并打印结果
函数将排序完成的切片返回,主函数输出最终结果[20,30]
二、时间复杂度分析
设输入数组长度为n(题目约束 1 ≤ n ≤ 100),去重后不同灯泡数量最多为100(灯泡编号1~100)。
1. 第一次遍历输入数组:遍历n个元素,单次map读写O(1),总耗时 O(n);
2. 遍历哈希字典筛选亮灯:最多遍历100个键,固定常数 O(1);
3. 排序结果切片:最多100个元素,常数规模排序 O(1);
整体时间复杂度:O(n)
常数操作不影响量级,核心耗时仅为遍历输入数组。
三、额外空间复杂度分析
额外开辟的存储空间:
1. map字典cnt:最多存储100组键值对(灯泡编号上限100),固定常量空间;
2. 结果切片ans:最多存放100个亮灯编号,固定常量空间;
所有额外空间不受输入数组长度n线性增长,存在固定上限。
整体额外空间复杂度:O(1)
Go完整代码如下:
package main
import (
"fmt"
"slices"
)
func toggleLightBulbs(bulbs []int) (ans []int) {
cnt := map[int]int{}
for _, i := range bulbs {
cnt[i] ^= 1
}
for i, c := range cnt {
if c > 0 {
ans = append(ans, i)
}
}
slices.Sort(ans)
return
}func main() {
bulbs := []int{10, 30, 20, 10}
result := toggleLightBulbs(bulbs)
fmt.Println(result)
}
![]()
Python完整代码如下:
# -*-coding:utf-8-*-
from collections import defaultdict
def toggle_light_bulbs(bulbs):
cnt = defaultdict(int)
for i in bulbs:
cnt[i] ^= 1 # 使用异或运算切换0/1状态
ans = [i for i, c in cnt.items() if c > 0]
ans.sort()
return ans
def main():
bulbs = [10, 30, 20, 10]
result = toggle_light_bulbs(bulbs)
print(result)if __name__ == "__main__":
main()
![]()
C++完整代码如下:
std::vector toggleLightBulbs(const std::vector& bulbs) {
std::map cnt;
for (int i : bulbs) {
cnt[i] ^= 1; // 使用异或运算切换0/1状态
}
std::vector ans;
for (const auto& pair : cnt) {
if (pair.second > 0) {
ans.push_back(pair.first);
}
}
// map已经按键排序,但为了与Go代码行为一致(Go中map无序需要排序),这里再次排序确保
// 实际上从map遍历已经是有序的,这行可以省略,但保留以明确意图
std::sort(ans.begin(), ans.end());
return ans;
}
int main() {
std::vector bulbs = {10, 30, 20, 10};
std::vector result = toggleLightBulbs(bulbs);
for (size_t i = 0; i < result.size(); ++i) {
std::cout << result[i];
if (i < result.size() - 1) {
std::cout << " ";
}
}
std::cout << std::endl;return0;
}
![]()
我们相信人工智能为普通人提供了一种“增强工具”,并致力于分享全方位的AI知识。在这里,您可以找到最新的AI科普文章、工具评测、提升效率的秘籍以及行业洞察。 欢迎关注“福大大架构师每日一题”,发消息可获得面试资料,让AI助力您的未来发展。