题目地址:
https://leetcode.com/problems/bulb-switcher/
给定 n n n个灯泡,编号 1 ∼ n 1\sim n 1∼n,一开始都是关的。接下来做 n n n次操作,第 i i i次操作要将所有 i i i的倍数的灯泡由开变关或者由关变开。问最后开着的灯泡有多少个。
编号为 k k k的灯泡仍然开着当且仅当其有奇数个因子,而一个数有奇数个因子当且仅当其是完全平方数。所以相当于在问 1 ∼ n 1\sim n 1∼n里有多少个完全平方数,所以答案是 [ n ] [\sqrt n] [n]。代码如下:
public class Solution {
public int bulbSwitch(int n) {
return (int) Math.sqrt(n);
}
}
时空复杂度 O ( 1 ) O(1) O(1)。