【leetcode】5353. 灯泡开关 III( Bulb Switcher III)

---

题目描述

【leetcode】5353. 灯泡开关 III( Bulb Switcher 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] 的一个排列。

第一次解答

思路
等差数列。当打开第i盏灯时，若前面的灯全亮，则已打开的灯序号必定包含从[1, i]中的所有数字，所以可以通过判断所有打开的灯序号是否为等差数列前n项和，若是，则变蓝灯。
这里可以不计算等差数列，用一个额外的变量不断累加即可。

代码：

class Solution {
public:
    int numTimesAllBlue(vector<int>& light) {
        int result = 0;
        // 等差数列，记录当前元素以及左边元素之和，为等差数列即可
        int count = 0;
        int count_real = 0;
        for(int i=0; i<light.size(); ++i){
            count_real += light[i];
            count += i+1;
            // 如果灯全亮
            if(count == count_real){
                result++;
            }
        }
        return result;
    }
};

结果：

第二次解答

题目中表示- 1 <= n <= 5 * 10^4，累加时用int保存结果也不会导致溢出，但若数据量再大些就不一定了。
由于当打开第i盏灯时，若前面的灯全亮，则已打开的灯序号必定包含从[1, i]中的所有数字，那么就可以知道，若第i盏灯前面灯全亮，则前面打开的灯序号的最大值 <= i，我们可以仅仅保留最大值即可。

class Solution {
public:
    int numTimesAllBlue(vector<int>& light) {
        int max_index = 0;
        int result = 0;
        for(int i=0; i<light.size(); ++i){
            if(light[i]-1 > max_index){
                max_index = light[i]-1;
            }
            if(i >= max_index ){
                result++;
            }
        }
        return result;
    }
};

结果：

相关/参考链接