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.

巴特西

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

最新文章

热门文章