1052. 爱生气的书店老板

转载请声明地址四元君

1052. 爱生气的书店老板

题目难度 Medium

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

思路

先假设该生气的时候生气，得到一个基本的值。接下来维护一个滑动窗口，来模拟这X分钟内不生气带来的新收益。

注意在边界条件时，返回的是所有customers的相加，不要简单的返回sum。

代码

class Solution {
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
        int sum=0,n=grumpy.size();
        for(int i=0;i<n;i++){
            if(grumpy[i]==0)    sum+=customers[i];
            if(grumpy[i]==1)    continue;
        }
        if(X>=n){
            sum=0;
            for(int i=0;i<n;i++){
                sum+=customers[i];
           
        }
            return sum;
        }
        int temp=0;
        for(int i=0;i<X;i++){
            if(grumpy[i]==1)    temp+=customers[i];
            if(grumpy[i]==0)    continue;
        }
        int pre=temp;
        for(int i=1;i<n-X+1;i++){
            int temp=pre;
            if(grumpy[i-1]==1)    temp-=customers[i-1];
            if(grumpy[i+X-1]==1)  temp+=customers[i+X-1];
            if(temp>temp)    temp=temp;
            pre=temp;
        }
        sum+=temp;
        return sum;
    }
};