leetcode算法题每日一练- 爱生气的书店老板

题目

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

分析

1. 首先根据题意，最大顾客数即为正常不生气的顾客数+X分钟的顾客数。
2. 所以有两个变量，即正常不生气的顾客数最大，还有一个是X分钟内顾客数最大。
3. 已知X是固定值，可以考虑滑动窗口方式。如示例1

customers = [1,0,1,2,1,1,7,5], 
grumpy = [0,1,0,1,0,1,0,1], 
X = 3

假设从customers[0]开始滑动，所以假设第一次店长就把X给用了。即grumpy数组中前三项都变成了0，因为店长这三分钟都不会生气，那么顾客最大数为平常不生气的顾客数=包含cunsomers[3]之后的顾客数-店长生气的顾客数 再加上X分钟的顾客数。
X分钟的顾客数直接计算即为1+0+1 = 2.
正常工作时间内的最大顾客数怎么算呢，一种就是先计算所有的顾客数即cunstomers[3]+cunstomers[4]+…cunstomers[7].
再减去剩下时间内店长生气的顾客数，观察到grumpy数组店长X分钟后共有三次生气，所以减去对应生气的顾客即为2+1+5.
所以就可得出最大顾客数= X顾客即为3+(平常最大顾客数16-生气的顾客数8)=11.
同理，再 从cunstomers[1]进行滑动，思想一样，不过要考虑使用X前的时间顾客计算

实现

 public static int maxSatisfied(int[] customers, int[] grumpy, int X) {
        // 最大顾客数
        int max = 0;
        // 遍历客户数组，注意防止下标越界
        for (int i = 0; i <= customers.length - X; i++) {
            // 单次遍历的最大顾客数
            int sum = 0;
            // 计算X分钟的最大顾客数，即窗口内的数据
            for (int j = 0; j < X; j++) {
                sum += customers[i + j];
            }
            //  计算使用过X后，剩余时间内的不生气用户的总数
            for (int j = i + X; j < customers.length; j++) {
                if (grumpy[j] == 0) {
                    sum += customers[j];
                }
            }
            //  计算使用过X前，剩余时间内的不生气用户的总数
            for (int j = 0; j < i; j++) {
                if (grumpy[j] == 0) {
                    sum += customers[j];
                }
            }
            max = Math.max(sum, max);
        }
        return max;
    }

总结

本题的空间复杂度为O(K)只使用了两次临时变量，时间复杂度为O(N*N)因为每滑动一次窗口，就重新统计了所有数组的值，贴下官方题解爱生气的书店老板