319. 灯泡开关
初始时有 n 个灯泡关闭。 第 1 轮,你打开所有的灯泡。 第 2 轮,每两个灯泡你关闭一次。 第 3 轮,每三个灯泡切换一次开关(如果关闭则开启,如果开启则关闭)。第 i 轮,每 i 个灯泡切换一次开关。 对于第 n 轮,你只切换最后一个灯泡的开关。 找出 n 轮后有多少个亮着的灯泡。
示例:
输入: 3
输出: 1
解释:
初始时, 灯泡状态 [关闭, 关闭, 关闭].
第一轮后, 灯泡状态 [开启, 开启, 开启].
第二轮后, 灯泡状态 [开启, 关闭, 开启].
第三轮后, 灯泡状态 [开启, 关闭, 关闭].
你应该返回 1,因为只有一个灯泡还亮着。
解题
第一轮:全亮
第二轮:2灭,4灭,6灭……
若n=10——1,2,5,10 四次转换,最后为灭;
若n=15——1,3,5,15 四次转换,最后为灭;
若n=20——1,2,4,5,10,20六次转换,最后为灭;
若n=25——1,5,25 三次转换,最后为亮;
所以挡因数个数为奇数个时,第n个灯泡为亮;
而只有完全平方数的因数个数为奇数个!
故统计小于n的完全平方数的个数即可;
即1,4,9,16,25……等即为亮着的灯;
class Solution {
public:
int bulbSwitch(int n) {
return sqrt(n);
}
};