1052. 爱生气的书店老板

2020-07-06

1.题目描述

今天，书店老板有一家店打算试营业 customers.length 分钟。每分钟都有一些顾客
（customers[i]）会进入书店，所有这些顾客都会在那一分钟结束后离开。

在某些时候，书店老板会生气。 如果书店老板在第 i 分钟生气，那么 grumpy[i] = 1，
否则 grumpy[i] = 0。 当书店老板生气时，那一分钟的顾客就会不满意，不生气则他们是满意的。

书店老板知道一个秘密技巧，能抑制自己的情绪，可以让自己连续 X 分钟不生气，但却只能使用一次。
请你返回这一天营业下来，最多有多少客户能够感到满意的数量。

2.题解

滑动窗口，这里窗口的大小是固定的，每次进行更新即可。

3.代码

class Solution {
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
        int len=customers.size();
        if (!len) return 0;
        int res=0,s1=0,s2=0;
        for (int i=0;i<len;i++){
            if (i<X) s1+=customers[i];
            else{
                if (grumpy[i]==0) s2+=customers[i];
            }
        }
        res=s1+s2;

        for (int i=0;i<len-X;i++){
            s1-=customers[i];
            s1+=customers[i+X];

            if (grumpy[i]==0){
                s2+=customers[i];
            }
            if (grumpy[i+X]==0){
                s2-=customers[i+X];
            }
            res=max(res,s1+s2);
        }
        return res;
    }
};