leetcode1052. 爱生气的书店老板(java)

爱生气的书店老板

难度 - 中等
原题链接 - 爱生气的书店老板

题目描述

有一个书店老板，他的书店开了 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 <= minutes <= n <= 2 * 1e4
0 <= customers[i] <= 1000
grumpy[i] == 0 or 1

滑动窗口

先把不生气的时候，求出来，再去计算连续不生气的时间内，找出最大值，就是滑动窗口最大值解题思路。
由于「技巧」只会将情绪将「生气」变为「不生气」，不生气仍然是不生气。
具体流程：
1.我们可以先将原本就满意的客户加入答案，同时将对应的 customers[i] 变为 0。
2.之后的问题转化为：在 customers 中找到连续一段长度为 minutes 的子数组，使得其总和最大。这部分就是我们应用技巧所得到的客户。

代码演示

class Solution {
    /**
     * 爱生气的书店老板
     * @param customers
     * @param grumpy
     * @param minutes
     * @return
     */
    public int maxSatisfied(int[] customers, int[] grumpy, int minutes) {
        int n = customers.length;
        int ans = 0;
        /**
         * 先计算出不生气时的顾客总量
         */
        for (int i = 0; i < n;i++){
            if (grumpy[i] == 0){
                ans += customers[i];
                customers[i] = 0;
            }
        }

        int max = 0,cur = 0;
        /**
         * 滑动窗口求出 minutes 范围内最大顾客量
         */
        for (int i = 0; i < n;i++){
            cur += customers[i];
            if (i >= minutes){
                cur -= customers[i - minutes];
            }
            max = Math.max(max,cur);
        }
        return ans + max;
    }
}

往期经典算法

leetcode - 75. 颜色分类

leetcode 475. 供暖器