爱生气的书店老板

题目:
今天,书店老板有一家店打算试营业 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.

解题思路:首先将grumpy[i] == 0时的客户满意度加起来,之后再利用滑动窗口将可以通过抑制情绪增加的客户满意度加起来即可

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        int sum = 0;
        // grumpy[i] == 0时的顾客必定是满意的
        for(int i = 0; i < customers.length; i++) {
            if(grumpy[i] == 0) {
                sum += customers[i];
            } else {
                // 将grumpy[i] == 1的顾客的满意度customers[i]转移到grumpy[i]中方便后续计算
                grumpy[i] = customers[i];
            }
        }
        
        int l = 0, r = 0;
        int save = 0;   // 当前窗口值
        // 确定窗口
        for(; r < X; r++) {
            save += grumpy[r];
        }
        
        int max = save;
        // 滑动窗口
        while(r < customers.length) {
            save += grumpy[r];
            save -= grumpy[l];
            max = Math.max(max, save);
            r++;
            l++;
        }
        
        return sum + max;
    }
}
posted on 2021-02-23 11:10  KobeSacre  阅读(44)  评论(0编辑  收藏  举报