Leetcode——爱生气的书店老板（maxSatisfied）

一.题目描述

今天，书店老板有一家店打算试营业 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的滑动窗口从数组customers的开始处往后移动，将窗口内的所有顾客以及窗口外所有的 grumpy[i]==0 的位置的顾客数相加，最后取所有结果中的最大值即可。

三.Python代码

#!/usr/bin/env python3
# encoding: utf-8
'''
@file: maxSatisfied.py
@time: 2020/7/13 0013 21:55
@author: Jack
@contact: jack18588951684@163.com
'''


class Solution:
    def maxSatisfied(self, customers, grumpy, X) -> int:
        n = len(customers)
        cur = 0  # 存放所有满意度
        for i in range(n):
            if i < X or grumpy[i] == 0:  # 滑动窗口的初始位置[0,1,2],滑动窗口内的顾客都给满意，滑动窗口外的grumpy=0的也给满意
                cur += customers[i]
        i = X  # 滑动窗口右边界
        ans = cur  # 目前的满意度
        while i < n:  # 滑动窗口依次往右移动
            if grumpy[i] == 1:  # 滑动窗口右边界，原来i在窗口外且grumpy=1的值没有被加入到cur中，现在i位置被滑动窗口覆盖到了，就要将对应的customer[i]值加入cur中
                cur += customers[i]
            if grumpy[i - X] == 1:  # 滑动窗口左边界，滑动窗口滑走后，i-X位置失去了滑动窗口的保护，若grumpy[i-X]=1,则滑动窗口滑走后，应该从cur减掉customers[i - X]
                cur -= customers[i - X]
            ans = max(ans, cur)  # 滑动窗口滑动过程中的cur值和ans（目前最大满意）进行比较，取较大者覆盖ans的值
            i += 1
        return ans


if __name__ == "__main__":
    s = Solution()
    customers = [1, 0, 1, 2, 1, 1, 7, 5]
    grumpy = [0, 1, 0, 1, 0, 1, 0, 1]
    X = 3
    res = s.maxSatisfied(customers, grumpy, X)
    print(res)

四.运行结果

16