题目:
有一个书店老板,他的书店开了 n
分钟。每分钟都有一些顾客进入这家商店。给定一个长度为 n
的整数数组 customers
,其中 customers[i]
是在第 i
分钟开始时进入商店的顾客数量,所有这些顾客在第 i
分钟结束后离开。
在某些时候,书店老板会生气。 如果书店老板在第 i
分钟生气,那么 grumpy[i] = 1
,否则 grumpy[i] = 0
。
当书店老板生气时,那一分钟的顾客就会不满意,若老板不生气则顾客是满意的。
书店老板知道一个秘密技巧,能抑制自己的情绪,可以让自己连续 minutes
分钟不生气,但却只能使用一次。
请你返回 这一天营业下来,最多有多少客户能够感到满意 。
算法;
方法一:滑动窗口
func maxSatisfied(customers []int, grumpy []int, minutes int) int {
satisfied := 0
// 统计满意的客户的满意度
for i := range customers {
if grumpy[i] == 0 {
satisfied = satisfied + customers[i]
}
}
ans := 0
// 对grumpy放置minutes的滑动窗口,一进一出,计算满意度,比较是否最大值
for left, right := 0, 0; right < len(grumpy); right ++ {
// 进来的时候是生气的话,咱消消火,增加满意度
if grumpy[right] == 1 {
satisfied = satisfied + customers[right]
}
if satisfied > ans {
ans = satisfied
}
// 达到minute窗口了,left右移,下一个时间可以进来窗口
if right - left + 1 == minutes {
// 在窗口内灭了火才有的满意度,现在离开窗口了,退还满意度
if grumpy[left] == 1 {
satisfied = satisfied - customers[left]
}
left ++
}
}
return ans
}
方法二:滑动窗口
func maxSatisfied(customers []int, grumpy []int, minutes int) int {
satisfied := 0
// 统计当前满意的客户总的满意度
for i := range customers {
if grumpy[i] == 0 {
satisfied = satisfied + customers[i]
customers[i] = 0
}
}
curSatisfied, max := 0, 0
// 统计通过老板控制情绪,最多能让多少不满意的客户满意
for i := range customers {
curSatisfied = curSatisfied + customers[i]
if i >= minutes {
curSatisfied = curSatisfied - customers[i - minutes]
}
if curSatisfied > max {
max = curSatisfied
}
}
return satisfied + max
}