319. 灯泡开关

最新推荐文章于 2024-10-02 23:46:40 发布

CS-Bob

最新推荐文章于 2024-10-02 23:46:40 发布

阅读量141

点赞数

分类专栏：算法文章标签：算法 leetcode 数据结构面试 java

本文链接：https://blog.csdn.net/bo123_/article/details/105525037

版权

算法专栏收录该内容

6 篇文章 0 订阅

订阅专栏

灯泡开关

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

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/bulb-switcher

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

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

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

模拟灯泡开关的过程的话，会因为输入的值太大，而超时。

根据网上的提示，我们使用平方数亮着的原理来解决，只有按了奇数次开关的灯泡才能打开。

第1轮：1,2,3,4,5,6,7,8,9，…

第2轮：2,4,6,8,10,12，…

第3轮：3,6,9,12,15,18，…

第j轮：j，2j,3j,4j,5j，… (int)sqrt(j)

…

第n轮：n

根据以上规律，我们可以知道

第1盏灯：第1 轮按下

第10盏灯：第1,2,5,10轮按下

第20盏灯：第1,2,4,5,10,20轮按下

第25盏灯：第1,5,25轮按下

所有的非平方数，都存在偶数个因子，因为P=AB，而平方数，存在奇数个因子，除了P=AB外，还有P=AA；

所以我们只需要找出n的所有平方数即可。

参考代码：

class Solution {
    public int bulbSwitch(int n) {

        int count = 0;
        for (int i = 1; i <= n; i++){
            if (i == (int)Math.sqrt(i) * (int)Math.sqrt(i)){
                count++;
            }
        }
        return count;
    }
}

参考题解：https://leetcode-cn.com/problems/bulb-switcher/solution/wei-shi-yao-ping-fang-shu-yi-ding-shi-liang-zhao-d/