题意:
今天,书店老板有一家店打算试营业 customers.length 分钟。
每分钟都有一些顾客(customers[i])会进入书店,所有这些顾客都会在那一分钟结束后离开。
在某些时候,书店老板会生气。 如果书店老板在第 i 分钟生气,那么 grumpy[i] = 1,否则 grumpy[i] = 0。
当书店老板生气时,那一分钟的顾客就会不满意,不生气则他们是满意的。
书店老板知道一个秘密技巧,能抑制自己的情绪,可以让自己连续 X 分钟不生气,但却只能使用一次。
请你返回这一天营业下来,最多有多少客户能够感到满意的数量。
数据范围:
1 <= X <= customers.length == grumpy.length <= 20000
0 <= customers[i] <= 1000
0 <= grumpy[i] <= 1
解法:
考虑X天最多能让多少个生气的客户变得不生气,
双指针维护一个大小为X的窗口,对窗口内生气客户的数量取max即可.
答案就是:max+不生气的数量.
code:
class Solution {
public:
int maxSatisfied(vector<int>& c, vector<int>& g, int X) {
int n=c.size();
int tot=0;
for(int i=0;i<n;i++){
if(!g[i]){
tot+=c[i];
}
}
int ans=0;
int sum=0;
for(int i=0;i<n;i++){
if(g[i])sum+=c[i];
if(i>=X&&g[i-X])sum-=c[i-X];
ans=max(ans,sum);
}
return tot+ans;
}
};