leetcode,脑筋急转弯,灯泡开关问题

题目

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

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

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

我的复杂做法

class Solution {
public:
    int bulbSwitch(int n) {
        if(n == 0)
            return 0;
        int sum = 1;
        for(int i = 2; i<= n;i++)
        {
            if(sqrt(n)-(int)sqrt(n) == 0)
                sum++;
        }
        return sum;
    }
};

别人的简单做法

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

你可能感兴趣的:(算法,leetcode,脑筋急转弯)