LeetCode：1052. 爱生气的书店老板————中等

题目
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

---

解题思路

• 滑动窗口。
• 我们求解的答案可以分解成两部分：一部分是所有不生气的时间的顾客数量；另一部分是在滑动窗口X内因老板生气被赶走的最大顾客数量。

---

Code

class Solution:
    def maxSatisfied(self, customers: List[int], grumpy: List[int], X: int) -> int:
        n = len(customers)
        sum1 = 0
        for i in range(n):  #计算不生气的顾客数量
            sum1 += customers[i] * (1 - grumpy[i])
        cur_value = 0
        for i in range(X):   #初始化窗口
            cur_value += customers[i] * grumpy[i]
        sum2 = cur_value
        for i in range(X,n):  #计算窗口内的生气的顾客数量
            #如果进入窗口的元素是生气的顾客，则添加到窗口中。
            #如果离开窗口的元素是生气的顾客，需要减去窗口内生气的顾客的数量
            cur_value = cur_value + customers[i] * grumpy[i] - customers[i - X] * grumpy[i - X]
            sum2 = max(sum2,cur_value)
        return sum1 + sum2

---

运行结果