leetcode 1052.爱生气的书店老板(python)

今天，书店老板有一家店打算试营业 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.

思路：利用滑动窗口思想

class Solution:
    def maxSatisfied(self, customers: List[int], grumpy: List[int], X: int) -> int:
        res = 0
        n = len(customers)

        for i in range(X):
            res += customers[i]             #前X个数先作为一个窗口向后滑动

        for i in range(X,n):
            if grumpy[i] == 0:
                res += customers[i]

        max_res = res
        for i in range(1,n-X+1):
            if grumpy[i-1] == 1:             #新离开窗口的数
                res -= customers[i-1]
            if grumpy[i+X-1] == 1:           #新进入窗口的数
                res += customers[i+X-1]
            if res > max_res:
                max_res = res
        
        return max_res