今天,书店老板有一家店打算试营业 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
- 首先就是将grumpy[i]=0情况下customers[i]的值进行累加,因为老板的技巧是将grumpy[i]=1变为grumpy[i]=0,原本满足grumpy[i]=0的第i分钟在连续的X分钟(视为一个窗口)下也并不会影响最后的结果。我们需要的是原本是grumpy[i]=1的第i分钟在窗口内才会增加满意的数量。所有在第一个循环代码中将满足grumpy[i]=0的customers[i] 置为0。
- 在下面代码中计算的是老板使用了技巧后额外增加的数量。窗口内满足grumpy[i]=0的customers[i]在前面被置为0,所以加上后不会影响最后的结果。接下来就是计算窗口内的和并不断将窗口向右滑动,得到最大的窗口值并和之前的总和进行相加。
public int maxSatisfied(int[] customers, int[] grumpy, int X) {
int n = customers.length;
//当老板没有使用技巧的时候,一天营业客户满意数
int sum_before_grumpy = 0;
for(int i=0;i<n;i++){
//老板在第i分钟不生气时
if(grumpy[i]==0){
sum_before_grumpy += customers[i];
//将第i分钟的顾客数置为0,防止下面代码重复累加
customers[i] = 0;
}
}
//将连续的X分钟视为一个窗口,maxWin记录这个窗口的最大值
int maxWin = 0,curWin = 0;
for(int i=0,j=0;i<n;i++){
//这就体现了上述代码将customers[i]置为0的作用,本来第i分钟不生气,即使第i分钟在窗口内也不会影响结果
curWin += customers[i];
//当窗口的长度超过X分钟时,因为上一行代码已经加了第i分钟的客户数,所以需要减去。j++就是将窗口的左边向右移动一位
if(i-j+1 > X){
curWin -= customers[j++];
}
maxWin = Math.max(maxWin, curWin);
}
return sum_before_grumpy + maxWin;
}