已解答
中等
相关标签
相关企业
初始时有 n
个灯泡处于关闭状态。第一轮,你将会打开所有灯泡。接下来的第二轮,你将会每两个灯泡关闭第二个。
第三轮,你每三个灯泡就切换第三个灯泡的开关(即,打开变关闭,关闭变打开)。第 i
轮,你每 i
个灯泡就切换第 i
个灯泡的开关。直到第 n
轮,你只需要切换最后一个灯泡的开关。
找出并返回 n
轮后有多少个亮着的灯泡。
示例 1:
输入:n = 3 输出:1 解释: 初始时, 灯泡状态 [关闭, 关闭, 关闭]. 第一轮后, 灯泡状态 [开启, 开启, 开启]. 第二轮后, 灯泡状态 [开启, 关闭, 开启]. 第三轮后, 灯泡状态 [开启, 关闭, 关闭]. 你应该返回 1,因为只有一个灯泡还亮着。
示例 2:
输入:n = 0 输出:0
示例 3:
输入:n = 1 输出:1
提示:
-
0 <= n <= 109
刚开始呢,蠢蠢的我当然想到了一个笨但是有效的作法,是能做出来,但超出了时间限制
int bulbSwitch(int n) { int cnt = 0; int *a = (int *)malloc((n + 1) * sizeof(int));//给数组在堆中申请内存 for (int h = 1; h < n + 1; h++)//把第一部全开省略,用1来表示亮 a[h] = 1; for (int i = 2; i <= n; i++)//外层表示轮数 { for (int j = i; j <= n; j += i)//内层表示跳到灯泡的反转 { a[j] = (-1) * a[j]; } } for (int h = 1; h <= n; h++)//计算一下>0也就是1的个数 { if (a[h] > 0) cnt++; } return cnt; }
然后,还有一个简化版的,不过要深刻理解
int bulbSwitch(int n) { return sqrt(n); }
第二个是不是很简单呢
这个优化的思路是:对于第 i 个灯泡,它会在第 i 轮被切换。而在第 i 轮之前,它会被切换的次数等于它的因子的个数。对于一个数 x,它的因子都是成对出现的,例如,12 的因子有 (1, 12), (2, 6), (3, 4)。唯一例外的是完全平方数,例如 4 的因子是 (1, 4), (2, 2)。因此,对于完全平方数,因子的个数是奇数,而其他数的因子个数是偶数。他刚开始是关的,如果因子是偶数的话,都是成对出现,一开一关,还是关的,但是如果因子是是奇数,也就是完全平方数,如4,其因子有1,2,4,最后肯定是关的
在本问题中,我们只需找到不超过 n 的完全平方数的个数,即 sqrt(n),这就是最终打开的灯泡数量。如n=16,开方为4,其包含的完全平方数有1,4,9,16;因此他返回4.
-
~~虽然我走得很慢,但绝不停止前行的脚步!加油!