一道单调栈的题目

一道单调栈的题目

背景

今天偶然遇到一道单调栈的题目,顺便复习下闲置已久的算法知识,题目的意思大致就是给你一个区间,找到子段和>=K的一个最小子段。

思路

子段问题的关键是前缀和,先求一下

int sum[50005]={0};
for(int i=0;i<len;i++) sum[i+1] = sum[i] + A[i];

紧接着,关键就是维持一个单调栈,下面解释一下什么是单调栈,先给一段代码

deque<int> dq; //双端队列数据结构
for(int i=0;i<=len;i++){
   while(!dq.empty() && sum[i]<=sum[dq.back()]){ //当前前缀小于
         dq.pop_back();
   }
   dq.push_back(i);
}

举个例子 [1,3,-4,5,-7] 前缀和[1,4,0,5,-2]
单调栈遍历过程中变化(记录下标)
i=0: 0
i=1: 单调上升 push 1,deque中为0,1
i=2: 不单调了,重新来 pop掉0和1,push 2 即deque中的数为下标2
i=3: 单调,push 3,deque中为2,3
i=4:不单调,pop掉2,3,push进4,即deque中为4

基于单调栈,如何找到子段和>=K的一个最小子段呢,如下所示:

class Solution {
public:
    int shortestSubarray(vector<int>& A, int K) {
        int len = A.size();
        int sum[50005]={0};
        // sum[0]=A[0];
        for(int i=0;i<len;i++) sum[i+1] = sum[i] + A[i];
        deque<int> dq;
        int ans = len+1;
        for(int i=0;i<=len;i++){
            while(!dq.empty() && sum[i]<=sum[dq.back()]){ //维持单调
                dq.pop_back();
            }
            while(!dq.empty() && sum[i]>=sum[dq.front()] + K){ //判断满足条件的单调栈,找到最小的
                // int t = dq.front();
                ans = min(ans, i-dq.front());
                dq.pop_front();
            }
            dq.push_back(i);
        }
        return ans<len+1?ans:-1;
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
单调是一种常用的数据结构,用于解决一类特定的问题,其中最常见的问题是找到数组中每个元素的下一个更大或更小的元素。在Codeforces编程竞赛中,单调经常被用于解决一些与数组相关的问题。 下面是单调的一般思路: 1. 创建一个空。 2. 从左到右遍历数组元素。 3. 对于每个元素,将其与顶元素进行比较。 - 如果当前元素小于等于顶元素,则将当前元素入。 - 如果当前元素大于顶元素,则将顶元素弹出,并将当前元素入。 4. 重复步骤3,直到遍历完所有元素。 这样,最后剩下的中元素就是没有下一个更大或更小元素的元素。在使用单调求解具体问题时,我们可以根据需要进行一些特定的操作。 例如,如果要找到一个数组中每个元素的下一个更大的元素,可以使用单调递减。具体操作如下: 1. 创建一个空和一个空结果数组。 2. 从左到右遍历数组元素。 3. 对于每个元素,将其与顶元素进行比较。 - 如果当前元素小于等于顶元素,则将当前元素入。 - 如果当前元素大于顶元素,则将顶元素弹出,并将其在结果数组中的位置记录为当前元素的下一个更大元素的索引。 4. 将当前元素入。 5. 重复步骤3和4,直到遍历完所有元素。 6. 结果数组中没有下一个更大元素的位置,可以设置为-1。 以下是一个使用单调递减求解下一个更大元素问题的示例代码: ```cpp #include <iostream> #include <stack> #include <vector> std::vector<int> nextGreaterElement(std::vector<int>& nums) { int n = nums.size(); std::vector<int> result(n, -1); std::stack<int> stack; for (int i = 0; i < n; i++) { while (!stack.empty() && nums[i] > nums[stack.top()]) { result[stack.top()] = i; stack.pop(); } stack.push(i); } return result; } int main() { std::vector<int> nums = {1,3, 2, 4, 5, 1}; std::vector<int> result = nextGreaterElement(nums); for (int i = 0; i < result.size(); i++) { std::cout << "Next greater element for " << nums[i] << ": "; if (result[i] != -1) { std::cout << nums[result[i]]; } else { std::cout << "None"; } std::cout << std::endl; } return 0; } ``` 以上代码将输出: ``` Next greater element for 1: 3 Next greater element for 3: 4 Next greater element for 2: 4 Next greater element for 4: 5 Next greater element for 5: None Next greater element for 1: None ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值