python leetcode 灯泡开关 III_1375

此题的思路在于亮灯的个数与最远亮灯的位置是否重合,
因此需要用一个变量记录当前亮灯里面最远亮灯的位置,
而亮灯的个数遵循, 每次循环, 亮灯个数加1.

class Solution(object):
    def numTimesAllBlue(self, light):
        """
        :type light: List[int]
        :rtype: int
        """
        ret = 0
        position = 0  # 记录亮灯里面最远的位置
        for i in range(len(light)):
            if position < light[i]:
                position = light[i]
            if i + 1 == position: # i + 1是亮灯的个数
                ret += 1
        return ret