Leetcode 319.灯泡开关
1 题目描述(Leetcode题目链接)
始时有 n 个灯泡关闭。 第 1 轮,你打开所有的灯泡。 第 2 轮,每两个灯泡你关闭一次。 第 3 轮,每三个灯泡切换一次开关(如果关闭则开启,如果开启则关闭)。第 i 轮,每 i 个灯泡切换一次开关。 对于第 n 轮,你只切换最后一个灯泡的开关。 找出 n 轮后有多少个亮着的灯泡。
输入: 3
输出: 1
解释:
初始时, 灯泡状态 [关闭, 关闭, 关闭].
第一轮后, 灯泡状态 [开启, 开启, 开启].
第二轮后, 灯泡状态 [开启, 关闭, 开启].
第三轮后, 灯泡状态 [开启, 关闭, 关闭].
你应该返回 1,因为只有一个灯泡还亮着。
2 题解
第 i i i个灯泡如果是亮着的,则它应该被按下奇数次。判断第 i i i下灯泡会被按几次就相当于判断 i i i有多少个因数,例如 8 8 8的因数有 1 , 2 , 4 , 8 1,2,4,8 1,2,4,8,所以 8 8 8号灯泡就是不亮的;而 4 4 4的因数有 1 , 2 , 4 1,2,4 1,2,4它有,所以 4 4 4号灯泡就是亮的。因此对于一个 n n n,我们只需要有几个小于 n n n的完全平方数,所以答案就是 ⌊ n ⌋ \lfloor \sqrt n \rfloor ⌊n⌋。
class Solution:
def bulbSwitch(self, n: int) -> int:
return math.floor(math.sqrt(n))