题目:
今天,书店老板有一家店打算试营业 customers.length 分钟。每分钟都有一些顾客(customers[i])会进入书店,所有这些顾客都会在那一分钟结束后离开。
在某些时候,书店老板会生气。 如果书店老板在第 i 分钟生气,那么 grumpy[i] = 1,否则 grumpy[i] = 0。 当书店老板生气时,那一分钟的顾客就会不满意,不生气则他们是满意的。
书店老板知道一个秘密技巧,能抑制自己的情绪,可以让自己连续 X 分钟不生气,但却只能使用一次。
请你返回这一天营业下来,最多有多少客户能够感到满意的数量。
示例:
输出: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数组中为1(表示生气)的元素置换为0(不生气),而原本为0的没有改变。
可以把customers数组中对应老板不生气的元素加起来,同时相应元素置为0,这样只需要找到customers数组中连续X个元素和中,最大的那个就可以了。
写法一:
class Solution {
public int maxSatisfied(int[] customers, int[] grumpy, int X) {
int max=0;
for(int i=0;i<grumpy.length;i++){
if(grumpy[i]==0){
max+=customers[i];
customers[i]=0;
}
}
int t=0;
for(int i=0;i<X;i++){
t+=customers[i];
}
int max1=t;
for(int i=X;i<customers.length;i++){
//这里是加上后一个,减去前一个就是下一个连续X的元素和了
t=t-customers[i-X]+customers[i];
max1=Math.max(t,max1);
}
return max1+max;
}
}
写法二:
class Solution {
public int maxSatisfied(int[] customers, int[] grumpy, int X) {
int max=0;
for(int i=0;i<grumpy.length;i++){
if(grumpy[i]==0){
max+=customers[i];
customers[i]=0;
}
}
int t=0,ans=0;
for(int i=0,j=0;i<customers.length;i++){
//这里是加上后一个,减去前一个就是下一个连续X的元素和了
t+=customers[i];
//这一步判断是为了计算出第一个连续X个元素的和t,之后同写法一
if(i>=X){
t-=customers[j++];
}
ans=Math.max(t,ans);
}
return ans+max;
}
}