Problem: 2398. 预算内的最多机器人数目
与239. 滑动窗口最大值类似
题意:
首先解读下题目,这道题题目也很难懂,我翻一下大致意思
给定两个长为n的数组,一个存放每台机器人的充电时间,另一个存放每台机器人的运行花销,k台机器人的花销为 m a x ( c h a r g e T i m e s ) + k ∗ s u m ( r u n n i n g C o s t s ) max(chargeTimes)+k∗sum(runningCosts) max(chargeTimes)+k∗sum(runningCosts) 。我们需要找到一个最大的连续区间使得其花销不大于给定的budget。
思路
解读完题目,我们很快就能发现 最大的连续区间 这几个关键字,那么我们就可以用到滑动窗口。
我先简单介绍下滑动窗口:
大白话就是向右扩展滑动窗口,并不断更新答案,当窗口内元素不满足要求了,这时候就需要从左端弹出元素直到窗口内满足要求。如此我们遍历的就都是满足要求的连续区间了。
例如:我们需要找到一个连续区间,它的和不大于target,此时我们向右遍历不断维护和sum和 ans,当sum > target时就需要从左端不断减去元素 sum -= nums[left++],直到sum不大于target
接下来是一段伪代码方便各位理解:
int ans =0, left = 0;
int sum = 0;
for(int i = 0; i < nums.size(); i++)
{
//向右拓展窗口
sum += nums[i];
while(sum > target)
{
//弹出左端元素,缩小窗口
sum -= nums[left++];
}
//更新ans
ans = max(ans, right - left + 1);
}
return ans;
单调队列
解决了获取最大连续序列的方法,但是我们又面临了一个新的问题,每个序列中最大的元素应该怎么获取,总不能每次都遍历下子序列,那样时间复杂度就爆了。
主角登场(单调队列)
大白话来解释这个数据结构,每次添加元素我都将最大的元素放到队列的最前端,仔细想想应该怎么做。
我们需要从以下几个方面考虑
- 如何将最大的元素维护在队列首部?
答: 添加新元素时,若队列尾部元素比当前元素小就应该pop出,其实小的元素我们根本也就没必要维护,我们需要的只是序列中最大的元素。 - 缩小窗口的时候单调队列应该怎么维护?
答:若left指向元素是队首元素,则队首元素出队,其余不用管。 - 虽然我们使用的基类是deque,但是只允许从队尾添加元素,两端皆可出队
class Solution {
public:
int maximumRobots(vector<int>& chargeTimes, vector<int>& runningCosts, long long budget) {
long long maxcharge = 0, sum = 0;
int left = 0, ans = 0;
//维护时传入下标可以简化后续的操作
deque<int> que;
for(int right = 0; right < runningCosts.size(); right++)
{
sum += runningCosts[right];
// 当队列前方有比当前元素小的就应该pop
while(!que.empty() && chargeTimes[right] > chargeTimes[que.back()])
{
que.pop_back();
}
que.push_back(right);
//维护窗口大小
while (!que.empty() && chargeTimes[que.front()] + (right - left + 1) * sum > budget) {
if (que.front() == left) {
que.pop_front();
}
sum -= runningCosts[left++];
}
//更新答案
ans = max(ans, right - left + 1);
}
return ans;
}
};