LeetCode: 1052. 爱生气的书店老板(中等)

题目描述

今天，书店老板有一家店打算试营业 customers.length 分钟。每分钟都有一些顾客（customers[i]）会进入书店，所有这些顾客都会在那一分钟结束后离开。

在某些时候，书店老板会生气。 如果书店老板在第 i 分钟生气，那么 grumpy[i] = 1，否则 grumpy[i] = 0。 当书店老板生气时，那一分钟的顾客就会不满意，不生气则他们是满意的。

书店老板知道一个秘密技巧，能抑制自己的情绪，可以让自己连续 X 分钟不生气，但却只能使用一次。

请你返回这一天营业下来，最多有多少客户能够感到满意的数量。

示例：
输入：customers = [1,0,1,2,1,1,7,5], grumpy = [0,1,0,1,0,1,0,1], X = 3
输出：16
解释：
书店老板在最后 3 分钟保持冷静。
感到满意的最大客户数量 = 1 + 1 + 1 + 1 + 7 + 5 = 16.

提示：

• 1 <= X <= customers.length == grumpy.length <= 20000
• 0 <= customers[i] <= 1000
• 0 <= grumpy[i] <= 1

题解

满意的数量 = 老板正常不生气的顾客满意数量(grumpy[i] 原本为0) + 老板使用秘密技巧获得顾客满意的数量(从 grumpy[i] 从1 -> 0)。

• 老板正常不生气的获得顾客的满意数量使用for循环计算grumpy[i]为0时的customer[i]值
• 老板使用秘密技巧获得顾客满意的数量采用滑动窗口计算，统计量为max
  a) 窗口扩大，将grumpy[i]为1的customer[i]加进来
  b) 若窗口大于X时，收缩窗口，减去grumpy[i-X]为1的customer[i-X]
  c) 若当前窗口内新增满意客户数count的值大于max,则更新max
  返回 base + max

class Solution {
     
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
     
        int max=0,base=0;//max记录通过抑制脾气改变的从不满意到满意的最大客户，base记录原先就是满意的客户数,count当前窗口内新增满意客户数
        for(int i=0,count=0;i<grumpy.size();i++)
        {
     
             if(grumpy[i]==0) base += customers[i];
             if(i < X && grumpy[i]==1)//滑动窗口小于X时 并且这时候会生气就加入count
                count += customers[i];
             // 滑动窗口右移中，(i - X, i] 这段就是滑动窗口
             else if(i >= X)
             {
     
                 //滑动窗口左端移出右端移进,移出时若老板生气,则减去此时顾客数,移进时若老板生气,则加上此时顾客数,
                 if(i-X >= 0 && grumpy[i-X]==1)
                 count -= customers[i-X];
                 if(grumpy[i]==1)
                 count += customers[i];
             }
             if(count > max) max = count;//更新max值
             
            
        }
        return base + max;
        
    }
};

复杂度分析
时间复杂度：O(N)，N 为数组长度。
空间复杂度：O(1)。

参考

c++ 滑动窗口
1052. 爱生气的书店老板【滑动窗口、JavaScript】
滑动窗口-爱生气的书店老板