每日一题:灯泡开关

文章讨论了一种问题,初始时n个灯泡关闭,每轮按一定规律开关灯泡。关键在于理解灯泡状态取决于其被操作次数的奇偶性,问题转化为求[1,n]中完全平方数的数量。给出了一种O(sqrt(n))的解决方案,利用完全平方数性质简化计算。
摘要由CSDN通过智能技术生成

初始时有 n 个灯泡处于关闭状态。第一轮,你将会打开所有灯泡。接下来的第二轮,你将会每两个灯泡关闭第二个。

第三轮,你每三个灯泡就切换第三个灯泡的开关(即,打开变关闭,关闭变打开)。第 i 轮,你每 i 个灯泡就切换第 i 个灯泡的开关。直到第 n 轮,你只需要切换最后一个灯泡的开关。

找出并返回 n 轮后有多少个亮着的灯泡。

示例 1:

输入:n = 3
输出:1 
解释:
初始时, 灯泡状态 [关闭, 关闭, 关闭].
第一轮后, 灯泡状态 [开启, 开启, 开启].
第二轮后, 灯泡状态 [开启, 关闭, 开启].
第三轮后, 灯泡状态 [开启, 关闭, 关闭]. 

你应该返回 1,因为只有一个灯泡还亮着。

提示:

  • 0 <= n <= 10^9

最容易想到的方法就是用hashmap存储每一个灯及其状态,而实质上可以退化用数组存储每个灯泡状态,模拟每一轮的操作,当前灯编号对第 i 轮取模,结果为0说明当前灯需要切换。

这种方法的时间复杂度是O(n^2)。是否有更简单的方法?

在上面模拟的过程中就能感受到一点端倪:在第 i 轮中,会变化的灯泡是编号能整除 i 的灯泡,也就是说,有因数 i 。

而一个灯泡是否打开,取决于它被操作的次数。初始为关,那么被切换次数为奇数的灯泡最终就是打开的。

也就是说,编号有奇数个因数的灯泡,是打开的。

那么问题就转化为,在[1,n]中,有几个数的因数是奇数个?

接下来的问题就是,什么样的数有奇数个因数?

因数是指可以整除另一个数的数。换句话说,如果 a 是 b 的因数,那么存在一个整数 c,使得 b = a × c。

也就是说,因数一般是成对出现的,如果出现一个数只有奇数个因数,那么一定是a = c了,也就是说,它是一个完全平方数,存在a使得 b = a^2。

问题进一步转化为,在[1,n]中,有多少个完全平方数?

设[1,n]中完全平方数的个数为k。那么,k个完全平方数必须是连续的整数,即:

1^{2} , 2^{2} ,......,k^{2}

 而:

1 <= k^2 <= n^2

 得到:

1 <= k <= \sqrt{n}

因此,[1,n]中完全平方数的个数为\sqrt{n}

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

这里+0.5是因为浮点数存储可能会丢失精度,举个例子:\sqrt{4} = 1.9999999,所以加一个小数可以保证得到正确的值。

  • 19
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值