LeetCode Algorithm 1052. 爱生气的书店老板

最新推荐文章于 2024-12-01 08:00:00 发布

原创最新推荐文章于 2024-12-01 08:00:00 发布 · 339 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode算法题专栏收录该内容

256 篇文章

订阅专栏

本文介绍了一种算法，通过滑动窗口技巧，计算在书店老板不生气的时间段内顾客总数，以及在X长度窗口内因老板生气而可能不满的顾客数，以优化顾客满意度。Python实现的Solution类展示了如何实现这个过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1052. 爱生气的书店老板

Ideas

找到一个时间长度为 X 的窗口包含最多原本因为老板生气而不满意的顾客（Sliding Window）

ans = 老板不生气时间内的顾客总数（customer1） + 窗口 X 内的原本不满意的顾客（customer2）

Python

class Solution:
    def maxSatisfied(self, customers: List[int], grumpy: List[int], X: int) -> int:
        n = len(customers)
        customer1 = sum(customers[i] * (1 - grumpy[i]) for i in range(n))
        customer2 = customer_x = sum(customers[i] * grumpy[i] for i in range(X))
        for i in range(X, n):
            customer_x = customer_x + customers[i] * grumpy[i] - customers[i - X] * grumpy[i - X]
            customer2 = max(customer2, customer_x)
        return customer1 + customer2