LeetCode 319. 灯泡开关（C++）

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

示例:

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

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

n轮后，各灯泡状态改变奇数次后，灯泡是亮的

class Solution {
public:
    int bulbSwitch(int n) 
    {
        vector<int> vec(n,0);
        for(int i=1;i<=n;i++)
        {
            for(int k=i-1;k<n;k+=i)
            {
                vec[k]++;
            }
        }
        int res=0;
        for(int i=0;i<n;i++)
        {
            if(1==vec[i]%2)
            {
                res++;
            }
        }
        return res;
    }
};

但是超时了，百度了一下，只要统计1~n有几个完全平方数就行了

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