这个题。。要我怎么说呢,HW实习和秋招二面都出的这道题,第二遍做了还是写了半天写出来。
有一个书店老板,他的书店开了 n 分钟。每分钟都有一些顾客进入这家商店。给定一个长度为 n 的整数数组 customers ,其中 customers[i] 是在第 i 分钟开始时进入商店的顾客数量,所有这些顾客在第 i 分钟结束后离开。
在某些时候,书店老板会生气。 如果书店老板在第 i 分钟生气,那么 grumpy[i] = 1,否则 grumpy[i] = 0。
当书店老板生气时,那一分钟的顾客就会不满意,若老板不生气则顾客是满意的。
书店老板知道一个秘密技巧,能抑制自己的情绪,可以让自己连续 minutes 分钟不生气,但却只能使用一次。
请你返回 这一天营业下来,最多有多少客户能够感到满意 。
思路:滑动窗口
不能直接两重循环,会超时。就维持一个left和一个right下标,然后
示例 1:
输入:customers = [1,0,1,2,1,1,7,5], grumpy = [0,1,0,1,0,1,0,1], minutes = 3
输出:16
解释:书店老板在最后 3 分钟保持冷静。
感到满意的最大客户数量 = 1 + 1 + 1 + 1 + 7 + 5 = 16.
class Solution {
public:
int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int minutes) {
int n = customers.size();
int a = 0;
for(int i=0;i<n;i++){
if(grumpy[i] == 0){
a += customers[i];
}
}
int left = 1;
int right = minutes;
int temp2=0;
for(int i=0;i<minutes;i++){
if(grumpy[i] == 1){
temp2 += customers[i];
}
}
int temp = temp2;
while( right<n){
if(grumpy[left-1] == 1){
temp2-=customers[left-1];
}
if(grumpy[right] == 1){
temp2+=customers[right];
}
left++;
right++;
temp = max(temp,temp2);
cout <<temp<<endl;
}
return a + temp;
}
};