python基础===100盏灯的问题

闪存里有人这样提问这样:

第一轮操作所有电灯,第二轮操作第2盏,第4盏开关,以此类推,第三轮改变编号为3的倍数的电灯,第3盏,第6盏,如果原来那盏灯是亮的,就熄灭它,如果原来是灭的,就点亮它,以此类推,直到第100轮

用python解决的思路如下:

灯泡有两种状态:开和关,用 -1代表关,用1代表开。每次操作后,状态上乘以-1。

这样问题就简单多了:

# on ---> 1     off ---> -1

dic = {k:-1 for k in range(1,101)}



def foo(n):
    for i in range(1,n+1):
        for x in dic:
            if x%i == False:
                dic[x] = int(dic[x])*-1

    return dic


num = len({k: v for k,v in foo(100).items() if v == 1})
print(f"There are {num} lights on.")

#There are 10 lights on.

 

posted @ 2018-09-12 16:54  botoo  阅读(1322)  评论(0编辑  收藏  举报