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.

 

思路分析：

连续X分钟，所以也可以用滑动窗口来解决。这是一个固定窗口滑动问题。

• 使用「秘密技巧」能得到的最终的顾客数 = 所有不生气时间内的顾客总数 + 在窗口 X 内使用「秘密技巧」挽留住的原本因为生气而被赶走顾客数。

因此，可以把题目分为以下两部分求解：

所有不生气时间内的顾客总数：使用 i 遍历[0, customers.length)，累加grumpy[i] == 0时的customers[i]。
在窗口 X 内因为生气而被赶走的顾客数：使用大小为 X 的滑动窗口，计算滑动窗口内的grumpy[i] == 1时的customers[i]，得到在滑动窗口内老板生气时对应的顾客数。最终求最大的赶走顾客数。

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        int n=customers.length;
        int init=0;
        //原本满意的客户
        for(int i=0;i<n;i++){
            init=init+customers[i]*(1-grumpy[i]);
        }
        int count=0;
        for(int i=0;i<X;i++){
            count=count+customers[i]*grumpy[i];
        }
        int maxCount=count;
        for(int i=X;i<n;i++){
            count=count+customers[i]*grumpy[i]-customers[i-X]*grumpy[i-X];
            maxCount=Math.max(maxCount,count);
        }
        return init+maxCount;
    }
}