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.

前缀和~
记录不算老板情绪的前缀和 $sum1$ 和算老板情绪的前缀和 $sum2$，遍历 $i\in [0,n-X]$，显然对每个 $i$，答案就是 $sum2.back()+sum1[i+X-1]-sum1[i-1]-(sum2[i+X-1]-sum2[i-1])$，这个公式计算的就是此位置往后连续 $X$ 分钟老板的情绪都不生气后增加的满意顾客数量，AC代码如下：

class Solution {
public:
    int maxSatisfied(vector<int> &customers, vector<int> &grumpy, int X) {
        vector<int> sum1, sum2;
        int n = customers.size(), ans = 0;
        for (int i = 0; i < n; i++) {
            if (sum1.empty()) sum1.emplace_back(customers[i]);
            else sum1.emplace_back(sum1.back() + customers[i]);
            if (sum2.empty()) sum2.emplace_back(grumpy[i] ? 0 : customers[i]);
            else sum2.emplace_back(sum2.back() + (grumpy[i] ? 0 : customers[i]));
        }
        for (int i = 0; i <= n - X; i++){
            ans = max(ans, sum2.back() + sum1[i + X - 1] - (i ? sum1[i - 1] : 0) - (sum2[i + X - 1] - (i ? sum2[i - 1] : 0)));
        }
        return ans;
    }
};