题目
初始时有 n 个灯泡关闭。 第 1 轮,你打开所有的灯泡。 第 2 轮,每两个灯泡你关闭一次。 第 3
轮,每三个灯泡切换一次开关(如果关闭则开启,如果开启则关闭)。第 i 轮,每 i 个灯泡切换一次开关。 对于第 n
轮,你只切换最后一个灯泡的开关。 找出 n 轮后有多少个亮着的灯泡。示例:
输入: 3
输出: 1
解释: 初始时, 灯泡状态 [关闭, 关闭, 关闭]. 第一轮后, 灯泡状态 [开启, 开启, 开启]. 第二轮后, 灯泡状态 [开启, 关闭, 开启]. 第三轮后, 灯泡状态 [开启, 关闭, 关闭].你应该返回 1,因为只有一个灯泡还亮着。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/bulb-switcher
解题思路
起始思路:
从后往前逆推,因为第i+1轮并不能影响到前面第i轮得到的结果,所以f(i+1) = (i+1) + f(i),把单独的第i+1次拿出来处理,一直递归直到i为0。
之后的思路:
因为如果第i轮能影响到第k个灯泡的结果,则说明该轮是k的因数,由因为必存在另一个因数j,使得i*j == k,所以,要使得灯泡开,只有存在i==j这一种情况,所以问题就变得异常简单了。
代码如下
public int bulbSwitch(int n) {
int res = 0;
for (int i = 1; i <= n; i++) {
int tmp = i*i;
if (tmp > n) {
break;
}
res++;
}
return res;
}