【一看就会一写就废 指间算法】爱生气的书店老板—— 滑动窗口

指尖划过的轨迹，藏着最细腻的答案~

题目：

有一个书店老板，他的书店开了 n 分钟。每分钟都有一些顾客进入这家商店。给定一个长度为 n 的整数数组 customers ，其中 customers[i] 是在第 i 分钟开始时进入商店的顾客数量，所有这些顾客在第 i 分钟结束后离开。

在某些分钟内，书店老板会生气。 如果书店老板在第 i 分钟生气，那么 grumpy[i] = 1，否则 grumpy[i] = 0。

当书店老板生气时，那一分钟的顾客就会不满意，若老板不生气则顾客是满意的。

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

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

示例 1：

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

示例 2：

输入：customers = [1], grumpy = [0], minutes = 1
输出：1

提示：

$n == customers.length == grumpy.length$
$1 \leq minutes \leq n \leq 2 \times 10^4$
$0 \leq customers[i] \leq 1000$
$grumpy[i] == 0$ or $1$

分析：

看到题目老板可以连续minutes 分钟不生气，但却只能使用一次，我们很容易可以想到使用定长滑动窗口，在窗口minutes中寻找最大和。

这样就行了吗？如果这个最大和正好在老板不生气的时候是不是就不一定符合题意了。为了解决这个问题，我们可以将最大和改为寻找生气时的最大和maxSum，这可以使minutes发挥最大的作用。

另外，我们在遍历过程中需要累加上老板不生气时的人数，最终加上maxSum即为答案。

AC代码：

class Solution {
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int minutes) {
        int ans = 0, maxSum = 0, sum = 0;
        int n = customers.size();

        for (int i = 0; i < n; i++) {
            // 进入窗口
            if (grumpy[i]) {
                sum += customers[i];
            } else {
                ans += customers[i];
            }

            int left = i - minutes + 1;
            if (left < 0) {
                continue;
            }

            // 更新答案
            maxSum = max(maxSum, sum);

            // 离开窗口
            int out = customers[left];
            if (grumpy[left]) {
                sum -= out;
            }
        }

        return ans + maxSum;
    }
};