计蒜客开关灯问题
计蒜客上的题目,我用的是python语言,我遇到的问题是超时,在网上寻求帮助后仍无法解决,现记录一下。
题目如下:
假设有 NNN 盏灯(NNN 为不大于 500050005000 的正整数),从 111 到 NNN 按顺序依次编号,初始时全部处于开启状态;有 MMM 个人(MMM 为不大于 NNN 的正整数)也从 111 到 MMM 依次编号。第一个人(111 号)将灯全部关闭,第二个人(222 号)将编号为 222 的倍数的灯打开,第三个人(333 号)将编号为 333 的倍数的灯做相反处理(即,将打开的灯关闭,将关闭的灯打开)。依照编号递增顺序,以后的人都和 333 号一样,将凡是自己编号倍数的灯做相反处理。请问:当第 MMM 个人操作之后,哪几盏灯是关闭的,按从小到大输出其编号,其间用逗号间隔。
我的代码:
初始代码:比较通俗易懂,完全按照题目要求
a =input().split()
st=int(a[1])
ac=int(a[0])
light=[]
list=[]
for i in range(ac):
if (i+