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; } };