初始时有 n
个灯泡处于关闭状态。第一轮,你将会打开所有灯泡。接下来的第二轮,你将会每两个灯泡关闭第二个。
第三轮,你每三个灯泡就切换第三个灯泡的开关(即,打开变关闭,关闭变打开)。第 i
轮,你每 i
个灯泡就切换第 i
个灯泡的开关。直到第 n
轮,你只需要切换最后一个灯泡的开关。
找出并返回 n
轮后有多少个亮着的灯泡。
示例 1:
输入:n = 3 输出:1 解释: 初始时, 灯泡状态 [关闭, 关闭, 关闭]. 第一轮后, 灯泡状态 [开启, 开启, 开启]. 第二轮后, 灯泡状态 [开启, 关闭, 开启]. 第三轮后, 灯泡状态 [开启, 关闭, 关闭]. 你应该返回 1,因为只有一个灯泡还亮着。
提示:
0 <= n <= 10^9
最容易想到的方法就是用hashmap存储每一个灯及其状态,而实质上可以退化用数组存储每个灯泡状态,模拟每一轮的操作,当前灯编号对第 i 轮取模,结果为0说明当前灯需要切换。
这种方法的时间复杂度是O(n^2)。是否有更简单的方法?
在上面模拟的过程中就能感受到一点端倪:在第 i 轮中,会变化的灯泡是编号能整除 i 的灯泡,也就是说,有因数 i 。
而一个灯泡是否打开,取决于它被操作的次数。初始为关,那么被切换次数为奇数的灯泡最终就是打开的。
也就是说,编号有奇数个因数的灯泡,是打开的。
那么问题就转化为,在[1,n]中,有几个数的因数是奇数个?
接下来的问题就是,什么样的数有奇数个因数?
因数是指可以整除另一个数的数。换句话说,如果 a 是 b 的因数,那么存在一个整数 c,使得 b = a × c。
也就是说,因数一般是成对出现的,如果出现一个数只有奇数个因数,那么一定是a = c了,也就是说,它是一个完全平方数,存在a使得 b = a^2。
问题进一步转化为,在[1,n]中,有多少个完全平方数?
设[1,n]中完全平方数的个数为k。那么,k个完全平方数必须是连续的整数,即:
而:
得到:
因此,[1,n]中完全平方数的个数为。
class Solution {
public:
int bulbSwitch(int n) {
return sqrt(n+0.5);
}
};
这里+0.5是因为浮点数存储可能会丢失精度,举个例子:,所以加一个小数可以保证得到正确的值。