LeetCode 爱生气的老板
爱生气的老板
今天,书店老板有一家店打算试营业 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
这道题很明显就是用滑动窗口来进行解题。
下面就是实现代码
class Solution {
public:
int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int minutes) {
//思路:滑动窗口
int ans = 0;
int size = customers.size();
//首先将老板不生气时的顾客相加
for(int i=0;i<size;i++){
if(grumpy[i]==0){
ans += customers[i];
}
}
//然后进行滑动窗口部分
//由于grumpy[i]==0不生气,grumpy[i]==1时生气,我们现在就是找生气数量最多的区间
//下面问题转换成customer[i]*grumpy[i]
//分两个部分进行相加,1. 当长度小于窗口长度,右指针直接相加。2. 当长度大于窗口长度,左指针右移。
int best = 0;
int cur = 0;
for(int i=0,j=0;i<size;i++){
cur += grumpy[i]*customers[i];
if(i-j+1>minutes){
cur -= customers[j]*grumpy[j];
j++;
}
best = max(best,cur);
}
return best+ans;
}
};