leetcode1052爱生气的书店老板
2021/2/23
爱生气的书店老板题目链接
今天,书店老板有一家店打算试营业 customers.length 分钟。每分钟都有一些顾客(customers[i])会进入书店,所有这些顾客都会在那一分钟结束后离开。
在某些时候,书店老板会生气。 如果书店老板在第 i 分钟生气,那么 grumpy[i] = 1,否则 grumpy[i] = 0。 当书店老板生气时,那一分钟的顾客就会不满意,不生气则他们是满意的。
书店老板知道一个秘密技巧,能抑制自己的情绪,可以让自己连续 X 分钟不生气,但却只能使用一次。
请你返回这一天营业下来,最多有多少客户能够感到满意的数量。
- 思路
先计算不生气时的满意数量,并把它们customers的值置0,方便计算使用技巧时的数量。
再用滑动窗口(大小为X)来计算使用技巧时最大的满意量。
两者相加就是结果 - 代码
class Solution {
public:
int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
int sum = 0;
int n = customers.size();
//顾客最初满意量
for(int i = 0; i < n; i++){
if(grumpy[i] == 0){
sum += customers[i];
customers[i] = 0;
}
}
//滑动窗口计算额外的满意量
int r = 1, l = 0;
int sum_extra = customers[0], ans = customers[0];
while(r < n){
while(r-l < X){
sum_extra += customers[r];
r++;
}
ans = max(ans, sum_extra);
sum_extra -= customers[l];
l++;
}
return sum+ans;
}
};
- 这个题学到最绝的:参考评论区大佬的customers[I] = 0 实在是太妙了
- 滑动窗口模板直接使用