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