2398.预算内的最多机器人数目
https://leetcode.cn/problems/maximum-number-of-robots-within-budget/
2398. 预算内的最多机器人数目
困难 │ 35 │ 37.4% 的 16.4K │ 提示
你有 n 个机器人,给你两个下标从 0 开始的整数数组 chargeTimes 和 runningCosts ,两者长度都为 n 。第 i 个机器人充电时间为 chargeTimes[i] 单位时间,花费 runningCosts[i] 单位时间运行。再给你一个整数 budget 。
运行 k 个机器人 总开销 是 max(chargeTimes) + k * sum(runningCosts) ,其中 max(chargeTimes) 是这 k 个机器人中最大充电时间,sum(runningCosts) 是这 k 个机器人的运行时间之和。
请你返回在 不超过 budget 的前提下,你 最多 可以 连续 运行的机器人数目为多少。
示例 1:
│ 输入:chargeTimes = [3,6,1,3,4], runningCosts = [2,1,3,4,5], budget = 25
│ 输出:3
│ 解释:
│ 可以在 budget 以内运行所有单个机器人或者连续运行 2 个机器人。
│ 选择前 3 个机器人,可以得到答案最大值 3 。总开销是 max(3,6,1) + 3 * sum(2,1,3) = 6 + 3 * 6 = 24 ,小于 25 。
│ 可以看出无法在 budget 以内连续运行超过 3 个机器人,所以我们返回 3 。
示例 2:
│ 输入:chargeTimes = [11,12,19], runningCosts = [10,8,7], budget = 19
│ 输出:0
│ 解释:即使运行任何一个单个机器人,还是会超出 budget,所以我们返回 0 。
提示:
* chargeTimes.length == runningCosts.length == n
* 1 <= n <= 5 * 10^4
* 1 <= chargeTimes[i], runningCosts[i] <= 10^5
* 1 <= budget <= 10^15
// @leet start
func maximumRobots(chargeTimes, runningCosts []int, budget int64) (ans int) {
sum := int64(0)
left := 0
q := []int{}
// 枚举区间右端点 right,计算区间左端点 left 的最小值
for right, t := range chargeTimes {
// 及时清除队列中的无用数据,保证队列的单调性
for len(q) > 0 && t >= chargeTimes[q[len(q)-1]] {
q = q[:len(q)-1]
}
q = append(q, right)
sum += int64(runningCosts[right])
// 如果左端点 left 不满足要求,就不断右移 left
for len(q) > 0 && int64(chargeTimes[q[0]])+int64(right-left+1)*sum > budget {
// 及时清除队列中的无用数据,保证队列的单调性
if q[0] == left {
q = q[1:]
}
sum -= int64(runningCosts[left])
left++
}
ans = max(ans, right-left+1)
}
return
}
// @leet end