[数据结构与算法]Leetcode.1052. 爱生气的书店老板---滑动窗口维护集合

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

题解：

• 我们先求出老板正常不生气时感到满意的客户数量，在此基础上在求出哪一段时间老板使用“技能”可以使得最多的不满意的客户感到满意即可 。
• 前面的过程使用一次for遍历+判断即可，后面的过程我们可以使用滑动窗口进行维护一个集合来解决。
• 由于我们将问题分解为了两部分求解，因此需要注意维护滑动窗口性质的集合时，若此时元素对应grumpy[i]==0，那么实际装入集合的是0（因为我们前面遍历已经算过了）。且每次我们计算集合内元素的和时可以不用每次都循环，我们可以使用记忆化的思想来简化此过程，使得时间复杂度降低。

代码：

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int minutes) {
        int sum = 0;
        for(int i=0;i<customers.length;i++){
            if(grumpy[i]==0){
                sum+=(customers[i]);
            }
        }

        ArrayList<Integer> list = new ArrayList<>();
        int temp = 0;
        for(int i=0;i<customers.length;i++){
            if(list.size()<minutes){
                if(grumpy[i]==0)
                    list.add(0);
                else
                    list.add(customers[i]);
            }
            else{
                int listSum = 0;
                for(int j=0;j<list.size();j++){
                    listSum=listSum+(Integer)(list.get(j));
                }
                temp = Math.max(temp,listSum);
                list.remove(0);
                i--;
            }
        }
        int listSum = 0;
        for(int j=0;j<list.size();j++){
            listSum+=(Integer)(list.get(j));
        }
        temp = Math.max(temp,listSum);
        
        sum+=temp;      
        return sum;
    }
}