题目描述 爱生气的书店老板
今天,书店老板有一家店打算试营业 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.
解题思路
- 先计算所有老板没有生气时候的客气
- 计算从0-X老板不生气增加的客人数
- 开始滑动窗口,当前是0-X,下一时候是1-X+1,计算i-i+X窗口内老板不生气增加的客人数,并且记录最大值
代码
class Solution {
public:
int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
int res=0;
int maxx = 0;
for(int i=0;i<customers.size();i++){
res += customers[i] * (1-grumpy[i]);
}
for(int i=0;i<X;i++){
maxx += customers[i] * grumpy[i];
}
int last = maxx;
for(int i=X;i<customers.size();i++){
int right = customers[i] * grumpy[i];
int left = customers[i-X] * grumpy[i-X];
last = right - left + last;
maxx = max(maxx, last);
}
return maxx + res;
}
};