第34届IMO第5题组合操作的模拟

冯涛20

于 2024-12-19 17:31:40 发布

阅读量107

点赞数 2

文章标签： python 算法

本文链接：https://blog.csdn.net/weixin_66206430/article/details/144590703

版权

一个经典拉灯游戏，但是操作次数是随着状态空间迅速增长的，大约为O(n^2*2^n)，不过在灯的数量为2的幂次或者2的幂次加1时却只有O(n^2)。

# IMO 34.6模拟

num = int(input("请输入数量："))
lamp = [1] * num
print("初始状态：", lamp)
lamp[0] = 0
print("第1步：", lamp)
step = 1
while True:
    stop = True
    index = step % num
    if lamp[index-1] == 1:
        if lamp[index] == 0:
            lamp[index] = 1
        else:
            lamp[index] = 0
    step += 1
    print(f"第{step}步：", lamp)
    for i in range(0, num):
        if lamp[i] == 0:
            stop = False
    if stop:
        print("成功终止！")
        break