901. 股票价格跨度「单调栈」

我特么服了,开始看题目还以为是在之前区间内找一个最大的连续天数。
没想到特么是需要连着当天的最长区间。

思路

因为是从今天开始往前走,找到一个大于它的价格即停止搜索,即往前找到第一个大于当前price的数,容易想到使用单调栈解决。
因为我们需要找到前面区间内第一个大于等于当前price的值,所以可以维护一个单调递减的栈。

求解找到第一个大于或小于当前值的相关问题,可以考虑使用单调栈来解决。

算法设计如下:

  1. 初始化栈s,以及当前价格所在位置下标curIdx(第几天)
  2. 调用next:新元素price进入后,先将curIdx++,再将栈顶小于等于price的元素全部出栈,直到栈为空或栈顶元素大于price
  3. 计算区间长度:为计算区间长度,每次元素入栈都需要将下标同时入栈。
    1. 区间左边界l:若此时栈为空则l = 0;否则令l为栈顶元素下标
    2. 区间右边界r:即为当前下标curIdx

此时区间长度为(l, r]中的元素个数,即为l-r

代码如下:

class StockSpanner {
public:
    StockSpanner() {
        curIdx = 0;
    }

    int next(int price) {
        while(!s.empty()){
            if (s.top().first <= price){
                s.pop();
            } else {
                break;
            }
        }
        int l = s.empty() ? 0 : s.top().second;
        s.push({price, ++curIdx});
        return curIdx - l;
    }

private:
    stack<pair<int, int>> s;
    int curIdx;
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值