【Leetcode】319. 灯泡开关

最新推荐文章于 2022-02-16 21:26:34 发布

EasonWEZ

最新推荐文章于 2022-02-16 21:26:34 发布

阅读量686

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： Leetcode

本文链接：https://blog.csdn.net/wez031113/article/details/83108089

LeetCode 专栏收录该内容

139 篇文章

订阅专栏

本文探讨了一道经典的灯泡开关谜题，通过n轮操作后计算亮着的灯泡数量。介绍了直观的暴力求解方法及超时问题，最终揭示了平方根规律的巧妙解法，提供了一个简洁高效的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

初始时有 n 个灯泡关闭。第 1 轮，你打开所有的灯泡。第 2 轮，每两个灯泡你关闭一次。第 3 轮，每三个灯泡切换一次开关（如果关闭则开启，如果开启则关闭）。第 i 轮，每 i 个灯泡切换一次开关。对于第 n 轮，你只切换最后一个灯泡的开关。找出 n 轮后有多少个亮着的灯泡。

示例:

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

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

解题思路：

我这题是试出来的，自己的代码超时了，但是试了几个测试用例，找到了规律。。。（^_^）

找到的解释的比较通俗易懂的博客——请参考。

AC代码：

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

TLE代码：

class Solution {
public:
    int bulbSwitch(int n) {
        int res = 0;
        vector<int> blub(n, -1);
        for (int i = 0; i < n; i++)
        {
            for (int j = i; j < n; j = j + i + 1)
            {
                blub[j] = -1 * blub[j];
            }
        }
        for (int i = 0; i < n; i++)
        {
            if (blub[i] == 1)
                res++;
        }
        return res;
    }
};