爱生气的书店老板

1052. 爱生气的书店老板

难度中等79

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

滑动窗口

首先预处理数组，只统计生气时候的人的数量，然后使用滑动窗口计算当前窗口内的和，在窗口滑动的过程中记录窗口中的最大和与左右边界

最后再遍历一次数组，计算满意的顾客的数量

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        int n=customers.length;
        
        // 预处理数组，只使用书店老板生气时候的人的数量
        int []temp=new int[n];
        for(int i=0;i<n;i++){
            if(grumpy[i]==1){
                temp[i]=customers[i];
            }else{
                temp[i]=0;
            }
        }

        int max=0;
        int c=0;
        // 最大值时的左右边界
        int maxLIndex=0,maxRIndex=0;
        // 滑动窗口边界
        int l=0,r=0;
        while(r<n){
            c+=temp[r];
            // 滑动窗口的不能继续扩大边界
            if(r-l+1>=X){
                if(c>max){
                    max=c;
                    maxLIndex=l;
                    maxRIndex=r;
                }
                c-=temp[l];
                l++;
            }
            r++;
        }
        
        int res=0;
        // 根据找到的边界计算结果值
        for(int i=0;i<n;i++){
            if(i>=maxLIndex&&i<=maxRIndex){
                res+=customers[i];
            }else{
                if(grumpy[i]==0){
                    res+=customers[i];
                }
            }
        }
        return res;
    }
}

官方题解滑动窗口(优雅)

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        int total = 0;
        int n = customers.length;
        for (int i = 0; i < n; i++) {
            if (grumpy[i] == 0) {
                total += customers[i];
            }
        }
        int increase = 0;
        for (int i = 0; i < X; i++) {
            increase += customers[i] * grumpy[i];
        }
        int maxIncrease = increase;
        for (int i = X; i < n; i++) {
            increase = increase - customers[i - X] * grumpy[i - X] + customers[i] * grumpy[i];
            maxIncrease = Math.max(maxIncrease, increase);
        }
        return total + maxIncrease;
    }
}

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/grumpy-bookstore-owner/solution/ai-sheng-qi-de-shu-dian-lao-ban-by-leetc-dloq/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

赏

谢谢你请我吃糖果

微信