LeetCode-Python-1375. 灯泡开关 III (数组)

房间中有 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

 

发布了722 篇原创文章 · 获赞 105 · 访问量 19万+

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/104746472