数据结构-单调队列 LeetCode-2398. 预算内的最多机器人数目 1917分

黄昏月亮很忙

于 2024-10-10 10:17:16 发布

阅读量48

点赞数

分类专栏： LeetCode 文章标签：数据结构 leetcode 算法

本文链接：https://blog.csdn.net/2402_84970476/article/details/142814401

版权

LeetCode 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

class Solution:
    def maximumRobots(self, chargeTimes: List[int], runningCosts: List[int], budget: int) -> int:
        ans = 0
        left = 0
        s = 0
        q = deque()

        for right, (charge, cost) in enumerate(zip(chargeTimes, runningCosts)):
            while q and charge >= chargeTimes[q[-1]]:
                q.pop()
            q.append(right)
            s += cost

            while q and chargeTimes[q[0]] + s * (right - left + 1) > budget:
                if q[0] == left:
                    q.popleft()
                s -= runningCosts[left]
                left += 1
            
            ans = max(ans, right - left + 1)
        
        return ans