LeetCode-Python-1375. 灯泡开关 III （数组）

最新推荐文章于 2024-10-25 14:12:58 发布

暴躁老哥在线刷题

最新推荐文章于 2024-10-25 14:12:58 发布

阅读量995

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.csdn.net/qq_32424059/article/details/104746472

Leetcode 专栏收录该内容

821 篇文章

订阅专栏

本文探讨了在特定条件下，一系列按顺序点亮的灯泡如何全部变为蓝色的问题。通过分析灯泡点亮的顺序，文章提出了一种线性扫描算法，该算法能够高效地计算出所有灯泡都变为蓝色的时刻数目。

房间中有 n 枚灯泡，编号从 1 到 n，自左向右排成一排。最初，所有的灯都是关着的。

在 k 时刻（ k 的取值范围是 0 到 n - 1），我们打开 light[k] 这个灯。

灯的颜色要想变成蓝色就必须同时满足下面两个条件：

灯处于打开状态。
排在它之前（左侧）的所有灯也都处于打开状态。
请返回能够让所有开着的灯都变成蓝色的时刻数目。

示例 1：

输入：light = [2,1,3,5,4]
输出：3
解释：所有开着的灯都变蓝的时刻分别是 1，2 和 4 。
示例 2：

输入：light = [3,2,4,1,5]
输出：2
解释：所有开着的灯都变蓝的时刻分别是 3 和 4（index-0）。
示例 3：

输入：light = [4,1,2,3]
输出：1
解释：所有开着的灯都变蓝的时刻是 3（index-0）。
第 4 个灯在时刻 3 变蓝。
示例 4：

输入：light = [2,1,4,3,6,5]
输出：3
示例 5：

输入：light = [1,2,3,4,5,6]
输出：6

提示：

n == light.length
1 <= n <= 5 * 10^4
light 是 [1, 2, ..., n] 的一个排列。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/bulb-switcher-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

某一时刻所有亮灯都变蓝的标志为：

亮灯中的最大下标等于当前亮灯数量。

举例，对于输入light = [3,2,4,1,5]：

1. 亮灯数为1，最大下标3

2. 亮灯数为2，最大下标3

3. 亮灯数为3，最大下标4

4. 亮灯数为4，最大下标4，这是一个亮灯全为蓝的时刻

5. 亮灯数为5，最大下标5，这也是一个亮灯全为蓝的时刻

所以只需要线性扫描数组，再用一个变量maxx记录当前亮灯中的最大下标即可。

时间复杂度：O（N）

空间复杂度：O（1）

class Solution(object):
    def numTimesAllBlue(self, light):
        """
        :type light: List[int]
        :rtype: int
        """
        res = 0
        maxx = 0
        for i in range(len(light)):
            maxx = max(light[i] - 1, maxx)
            if i == maxx:
                res += 1
        return res