【每日一题】LeetCode 1052.爱生气的书店老板(数组、滑动窗口)
题目描述
书店老板的商店每天有不同数量的顾客进入。每分钟,老板可能或可能不会生气。如果老板生气,那一分钟的顾客就会不满意。老板知道一个秘密技巧,可以让自己在连续的 minutes
分钟内不生气,从而让所有顾客都满意。我们的任务是找出使用这个技巧后,一天中最多有多少顾客能够感到满意。
customers
:一个整数数组,表示每分钟进入商店的顾客数量。grumpy
:一个整数数组,表示老板每分钟是否生气,1
表示生气,0
表示不生气。minutes
:一个整数,表示老板可以使用技巧的连续分钟数。
示例
示例 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。
示例 2
- 输入:
customers = [1], grumpy = [0], minutes = 1
- 输出:
1
- 解释:老板不生气,所有顾客都感到满意。
思路分析
这个问题可以通过滑动窗口的方法来解决。首先,我们计算在没有使用技巧的情况下,顾客感到满意的总数量。然后,我们使用一个滑动窗口来模拟老板使用技巧的情况。窗口的大小是 minutes
分钟,我们计算在这个窗口内,如果老板不生气,顾客的数量。接着,我们移动窗口,每次移动都会更新窗口内的顾客数量,并记录下最大的顾客数量。最后,我们将未使用技巧时的顾客数量与使用技巧时的最大顾客数量相加,得到最终的结果。
代码实现
class Solution {
public int maxSatisfied(int[] customers, int[] grumpy, int minutes) {
int totalSatisfied = 0; // 记录老板不生气时的总顾客数量
int n = customers.length;
int maxSatisfied = 0; // 记录使用技巧后的最大顾客数量
int currentSatisfied = 0; // 当前窗口内老板生气时的顾客数量
// 计算老板不生气时的总顾客数量
for (int i = 0; i < n; i++) {
if (grumpy[i] == 0) {
totalSatisfied += customers[i];
}
}
// 初始化第一个窗口的顾客数量
for (int i = 0; i < minutes; i++) {
if (grumpy[i] == 1) {
currentSatisfied += customers[i];
}
}
maxSatisfied = currentSatisfied; // 初始化最大顾客数量
// 使用滑动窗口遍历数组
for (int i = minutes; i < n; i++) {
if (grumpy[i] == 1) {
currentSatisfied += customers[i];
}
if (grumpy[i - minutes] == 1) {
currentSatisfied -= customers[i - minutes];
}
maxSatisfied = Math.max(maxSatisfied, currentSatisfied);
}
// 返回最终的最大顾客数量
return totalSatisfied + maxSatisfied;
}
}