算法学习每日一题之2398. 预算内的最多机器人数目:滑动窗口+单调队列

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)+ksum(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;

单调队列

解决了获取最大连续序列的方法,但是我们又面临了一个新的问题,每个序列中最大的元素应该怎么获取,总不能每次都遍历下子序列,那样时间复杂度就爆了。

主角登场(单调队列)

大白话来解释这个数据结构,每次添加元素我都将最大的元素放到队列的最前端,仔细想想应该怎么做。

我们需要从以下几个方面考虑

  1. 如何将最大的元素维护在队列首部?
    答: 添加新元素时,若队列尾部元素比当前元素小就应该pop出,其实小的元素我们根本也就没必要维护,我们需要的只是序列中最大的元素。
  2. 缩小窗口的时候单调队列应该怎么维护?
    答:若left指向元素是队首元素,则队首元素出队,其余不用管。
  3. 虽然我们使用的基类是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;
        
        
    }
};
  • 17
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值