【经典算法题】灯泡开关

【经典算法题】灯泡开关

Leetcode 0319 灯泡开关

题目描述：Leetcode 0319 灯泡开关

分析

• 本题的考点：数学。

• 一个灯泡x最后是亮的$⟺$它被按奇数次$⟺$它的约数个数是奇数。

• 因此最后所有约数个数是奇数个的灯泡是亮的，比如5个灯泡被按5次，1、4灯泡就是亮的，因为1只有1这一个约数，4有1、2、4三个约数。

• 下面考虑约数个数为奇数的数据有什么特点：如果一个数N可以质因数分解为 $p_1^{{\alpha }_1}\ast p_2^{{\alpha }_2}\ast ...\ast p_k^{{\alpha }_k}$ ，则N的约数个数为 $({\alpha }_1+1)\ast ({\alpha }_2+1)\ast ...\ast ({\alpha }_k+1)$ 。则必须有${\alpha }_i+1$必须是奇数，所以${\alpha }_i$必须是偶数，因此N是完全平方数。

• 上面证明了如果一个数N的约数个数为奇数，则其是完全平方数。反过来也成立（例如$x=y^2$，只有约数y出现一次，其余都是成对出现的）。

• 因此本题的最后答案就是1~n中完全平方数的个数。

代码

• C++

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

• Java

class Solution {

    public int bulbSwitch(int n) {
        return (int) Math.sqrt(n);
    }
}

时空复杂度分析

• 时间复杂度：$O(1)$。

• 空间复杂度：$O(1)$。