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

  1. 输入: 3
  2. 输出: 1
  3. 解释:
  4. 初始时, 灯泡状态 [关闭, 关闭, 关闭].
  5. 第一轮后, 灯泡状态 [开启, 开启, 开启].
  6. 第二轮后, 灯泡状态 [开启, 关闭, 开启].
  7. 第三轮后, 灯泡状态 [开启, 关闭, 关闭].
  8. 你应该返回 1,因为只有一个灯泡还亮着。

暴力超时法

class Solution {
public:
    int bulbSwitch(int n) {

        if(n == 1){
            return 1;

        }
        vector<int> dp(n + 1, 0);
        for(int i = 1; i<= n;i++){
            for(int j = i; j<=n; j=j+i){
                dp[j] = dp[j] == 0 ? 1:0;
            }
        }
        return accumulate(dp.begin(), dp.end(),0);

    }
};

找规律法

class Solution {
public:
    int bulbSwitch(int n) {
        int count = 0;
        int index = 0;

        while(index < n){
            count++;
            index +=  2 * count + 1;
            cout<<index<<endl;
        }

        return count;

    }
};

因数法

标号为n的因子数的个数意味着灯泡操作的次数,次数为奇数则为亮的。只有平方数的因子数才为奇数。

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