爱生气的书店老板

题目描述

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.

提示：

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 maxSatisfied = 0;
        for(int i = 0;i<=customers.length-X;i++){
            int curSatisfied=0;
            for(int j = 0;j<i;j++){//连续不生气之前
                curSatisfied+=customers[j]*(1-grumpy[j]);
            }
            for(int j=i;j<=i+X-1;j++){//连续不生气
                curSatisfied+=customers[j];
            }
            for(int j=i+X;j<customers.length;j++){//连续不生气之后
                curSatisfied+=customers[j]*(1-grumpy[j]);
            }
            maxSatisfied=Math.max(maxSatisfied,curSatisfied);
        }
        return maxSatisfied;
    }
}

暴力解法思路很简单，只是在grumpy数组上滑窗长度为X的区间，作为连续不生气的时间区间。

 观察发现暴力解法每次都统计整个数组的和，但其实对于没有使用技巧的和，使用技巧后的和只是改变了长度为X的滑动窗口的部分和，因此可以先求出整个数组的和，再计算滑动窗口内翻转的部分和的最大值，于是就有了下面的优化解法。

优化解法

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        int allSatisNum = 0;
        for(int i=0;i<customers.length;i++){
            allSatisNum += customers[i]*(grumpy[i]^1);//统计没有使用秘密技巧时满意的人数总和
        }
        int curExtraNum = 0;
        int maxExtraNum = 0;
        for(int i=0;i<X;i++){//计算首个滑动窗口的翻转部分和
            curExtraNum += customers[i]*grumpy[i];
        }
        maxExtraNum = curExtraNum;
        for(int i=X;i<customers.length;i++){//滑动窗口
            curExtraNum += customers[i]*grumpy[i];//加上滑动窗口右移新添的
            curExtraNum -= customers[i-X]*grumpy[i-X];//减去滑动窗口右移抛弃的
            maxExtraNum = Math.max(curExtraNum,maxExtraNum);//更新翻转部分和的最大值
        }
        return allSatisNum + maxExtraNum;
    }
}

 进一步优化

仔细观察发现，上述解法对customers数组进行了两次互不干扰的遍历，因此，两次遍历可以合为一次。

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        int allSatisNum = 0;
        int curExtraNum = 0;
        int maxExtraNum = 0;
        for(int i=0;i<customers.length;i++){
            allSatisNum += customers[i]*(grumpy[i]^1);//统计没有使用秘密技巧时满意的人数总和
            if(i<X){//计算首个滑动窗口的翻转部分和
                curExtraNum += customers[i]*grumpy[i];
                maxExtraNum = curExtraNum;
            }else{
                curExtraNum += customers[i]*grumpy[i];//加上滑动窗口右移新添的
                curExtraNum -= customers[i-X]*grumpy[i-X];//减去滑动窗口右移抛弃的
                if(curExtraNum>maxExtraNum){
                    maxExtraNum = curExtraNum;//更新翻转部分和的最大值
                }
            }
        }
        return allSatisNum + maxExtraNum;
    }
}