题目描述
今天,书店老板有一家店打算试营业 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) + 老板使用秘密技巧获得顾客满意的数量(从 grumpy[i] 从1 -> 0)。
- 老板正常不生气的获得顾客的满意数量使用for循环计算grumpy[i]为0时的customer[i]值
- 老板使用秘密技巧获得顾客满意的数量采用滑动窗口计算,统计量为max
a) 窗口扩大,将grumpy[i]为1的customer[i]加进来
b) 若窗口大于X时,收缩窗口,减去grumpy[i-X]为1的customer[i-X]
c) 若当前窗口内新增满意客户数count的值大于max,则更新max
返回 base + max
class Solution {
public:
int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
int max=0,base=0;//max记录通过抑制脾气改变的从不满意到满意的最大客户,base记录原先就是满意的客户数,count当前窗口内新增满意客户数
for(int i=0,count=0;i<grumpy.size();i++)
{
if(grumpy[i]==0) base += customers[i];
if(i < X && grumpy[i]==1)//滑动窗口小于X时 并且这时候会生气就加入count
count += customers[i];
// 滑动窗口右移中,(i - X, i] 这段就是滑动窗口
else if(i >= X)
{
//滑动窗口左端移出右端移进,移出时若老板生气,则减去此时顾客数,移进时若老板生气,则加上此时顾客数,
if(i-X >= 0 && grumpy[i-X]==1)
count -= customers[i-X];
if(grumpy[i]==1)
count += customers[i];
}
if(count > max) max = count;//更新max值
}
return base + max;
}
};
复杂度分析
时间复杂度:O(N),N 为数组长度。
空间复杂度:O(1)。