每日算法系列【LeetCode 319】灯泡开关

最新推荐文章于 2022-03-02 10:32:36 发布

fe_lucifer

最新推荐文章于 2022-03-02 10:32:36 发布

阅读量160

点赞数

原文链接：https://zhuanlan.zhihu.com/p/111066645

版权

初始时有个灯泡关闭。第轮，你打开所有的灯泡。第轮，每两个灯泡你关闭一次。第轮，每三个灯泡切换一次开关（如果关闭则开启，如果开启则关闭）。第轮，每个灯泡切换一次开关。对于第轮，你只切换最后一个灯泡的开关。找出轮后有多少个亮着的灯泡。

示例1

输入：
3
输出：
1
解释：
初始时, 灯泡状态 [关闭, 关闭, 关闭].
第一轮后, 灯泡状态 [开启, 开启, 开启].
第二轮后, 灯泡状态 [开启, 关闭, 开启].
第三轮后, 灯泡状态 [开启, 关闭, 关闭]. 


你应该返回 1，因为只有一个灯泡还亮着。

首先有个灯泡，假设编号为到。第轮，所有编号是的倍数的灯泡被开关了一次。第轮，所有编号是的倍数的灯泡被开关了一次。类推下去，第轮，所有编号是的倍数的灯泡被开关了一次。

综上，对于编号为的灯泡来说，它最终被开关的次数取决于有几个因数。如果有奇数个因数，那么它最后就是开着的，否则就是关着的。

那么我们有一个定理：如果一个正整数有奇数个因数，那么它一定是完全平方数。

最浅显的证明就是，一个数的因数按照从小到大排个序，首尾两两一对之积一定等于。而如果因数只有奇数个，最中间一个因数只会出现一次，那么。

严格证明也不难，首先将质因数分解为：

那么的因数个数就是：

因为的因数个数是奇数，所以任意必定是奇数，即任意必定是偶数。

那么就可以写作：

这就证明了一定是一个完全平方数。

所以问题就转化为了求到之间有多少个完全平方数。答案就是。

在具体实现的时候，为了防止出现浮点数误差（比如算出来是，取整得到），我们可以计算的结果。

class Solution {
public:
    int bulbSwitch(int n) {
        return sqrt(n+0.5);
    }
};

推荐阅读

如果觉得文章不错，帮忙点个在看呗

关注