[LeetCode] Bulb Switcher 灯泡开关

 

There are n bulbs that are initially off. You first turn on all the bulbs. Then, you turn off every second bulb. On the third round, you toggle every third bulb (turning on if it's off or turning off if it's on). For the nth round, you only toggle the last bulb. Find how many bulbs are on after n rounds.

Example:

Given n = 3. 

At first, the three bulbs are [off, off, off].
After first round, the three bulbs are [on, on, on].
After second round, the three bulbs are [on, off, on].
After third round, the three bulbs are [on, off, off]. 

So you should return 1, because there is only one bulb is on.

 

这道题给了咱们n个灯泡，第一次打开全部的灯泡，第二次每两个更改灯泡的状态，第三次每三个更改灯泡的状态，以此类推，第n次每n个更改灯泡的状态。让咱们求n次后，全部亮的灯泡的个数。此题是CareerCup 6.6 Toggle Lockers 切换锁的状态。

那么咱们来看这道题吧，仍是先枚举个小例子来分析下，好比只有5个灯泡的状况，'X'表示灭，‘√’表示亮，以下所示：

初始状态：    X    X    X    X    X

第一次：      √    √    √    √    √

第二次：      √     X    √    X    √

第三次：      √     X    X    X    √

第四次：      √     X    X    √    √

第五次：      √     X    X    √    X

那么最后咱们发现五次遍历后，只有1号和4号灯泡是亮的，并且很巧的是它们都是平方数，是巧合吗，仍是其中有什么玄机。咱们仔细想一想，对于第n个灯泡，只有当次数是n的因子的以后，才能改变灯泡的状态，即n能被当前次数整除，好比当n为36时，它的因数有(1,36), (2,18), (3,12), (4,9), (6,6), 能够看到前四个括号里成对出现的因数各不相同，括号中前面的数改变了灯泡状态，后面的数又变回去了，等于灯泡的状态没有发生变化，只有最后那个(6,6)，在次数6的时候改变了一次状态，没有对应其它的状态能将其变回去了，因此灯泡就一直是点亮状态的。因此全部平方数都有这么一个相等的因数对，即全部平方数的灯泡都将会是点亮的状态。

那么问题就简化为了求1到n之间彻底平方数的个数，咱们能够用force brute来比较从1开始的彻底平方数和n的大小，参见代码以下：

 

解法一：

class Solution {
public:
    int bulbSwitch(int n) {
        int res = 1;
        while (res * res <= n) ++res;
        return res - 1;
    }
};

 

还有一种方法更简单，咱们直接对n开方，在C++里的sqrt函数返回的是一个整型数，这个整型数的平方最接近于n，即为n包含的全部彻底平方数的个数，参见代码以下：

 

解法二：

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

 

讨论：这道题有个follow up就是，若是咱们toggle的顺序不是1，2，3，4...，而是1，3，5，7...，或者是2，4，6，8... 的话，还怎么作？博主没有想出像解法二那样简便的方法，只是大概想了想，若是各位大神有更好的方法，请必定要在下方留言啊。博主想的是，好比对于1，3，5，7...，那么就是先把全部的灯点亮，而后关掉3，6，9，12，15...等的灯，而后toggle的是5，10，15...等等，而后再toggle的是7，14，21...，咱们发现，纯2的倍数的灯永远不会被改变，好比2，4，8，16... 这些灯状态不会变，有些灯只会变一次，好比3，6，9等，而有些灯会变两次，好比15（3x5），21（3x7），35（5x7）等，有些灯会变三次，好比105（3x5x7），那么咱们能够观察出规律了，toggle的次数跟奇数因子的数字有关（注意这里的奇数因子不包括1），只要有奇数个奇因子，那么灯就是灭的，只要有偶数个奇因子，那么灯就是亮的。

 

相似题目：

Bulb Switcher II

6.6 Toggle Lockers

 

参考资料：

https://leetcode.com/problems/bulb-switcher/discuss/77104/Math-solution..

https://leetcode.com/problems/bulb-switcher/discuss/77112/Share-my-o(1)-solution-with-explanation

 

LeetCode All in One 题目讲解汇总(持续更新中...)