forked from haoel/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathbulbSwitcher.cpp
44 lines (43 loc) · 1.79 KB
/
bulbSwitcher.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
// Source : https://leetcode.com/problems/bulb-switcher/
// Author : Calinescu Valentin
// Date : 2015-12-28
/***************************************************************************************
*
* 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.
*
***************************************************************************************/
/*
* Solution 1 - O(1)
* =========
*
* We notice that for every light bulb on position i there will be one toggle for every
* one of its divisors, given that you toggle all of the multiples of one number. The
* total number of toggles is irrelevant, because there are only 2 possible positions(on,
* off). We quickly find that 2 toggles cancel each other so given that the start position
* is always off a light bulb will be in if it has been toggled an odd number of times.
* The only integers with an odd number of divisors are perfect squares(because the square
* root only appears once, not like the other divisors that form pairs). The problem comes
* down to finding the number of perfect squares <= n. That number is the integer part of
* the square root of n.
*
*/
class Solution {
public:
int bulbSwitch(int n) {
return (int)sqrt(n);
}
};