901. 股票价格跨度

最新推荐文章于 2023-10-07 13:06:51 发布

empty_coder

最新推荐文章于 2023-10-07 13:06:51 发布

阅读量759

点赞数

分类专栏：单调栈文章标签：单调栈

本文链接：https://blog.youkuaiyun.com/emptycodervblog/article/details/82697717

版权

单调栈专栏收录该内容

1 篇文章

订阅专栏

传送门：https://leetcode-cn.com/contest/weekly-contest-101/problems/online-stock-span/

编写一个 StockSpanner 类，它收集某些股票的每日报价，并返回该股票当日价格的跨度。

今天股票价格的跨度被定义为股票价格小于或等于今天价格的最大连续日数（从今天开始往回数，包括今天）。

例如，如果未来7天股票的价格是 [100, 80, 60, 70, 60, 75, 85]，那么股票跨度将是 [1, 1, 1, 2, 1, 4, 6]。

示例：

输入：["StockSpanner","next","next","next","next","next","next","next"], [[],[100],[80],[60],[70],[60],[75],[85]]
输出：[null,1,1,1,2,1,4,6]
解释：
首先，初始化 S = StockSpanner()，然后：
S.next(100) 被调用并返回 1，
S.next(80) 被调用并返回 1，
S.next(60) 被调用并返回 1，
S.next(70) 被调用并返回 2，
S.next(60) 被调用并返回 1，
S.next(75) 被调用并返回 4，
S.next(85) 被调用并返回 6。

注意 (例如) S.next(75) 返回 4，因为截至今天的最后 4 个价格
(包括今天的价格 75) 小于或等于今天的价格。

提示：

调用 StockSpanner.next(int price) 时，将有 1 <= price <= 10^5。
每个测试用例最多可以调用 10000 次 StockSpanner.next。
在所有测试用例中，最多调用 150000 次 StockSpanner.next。
此问题的总时间限制减少了 50%。

分析：维护一个单调（递减）栈，栈中元素为股票跨度中的最大元素

时间复杂度： $O(N),N为调用次数$ 为调用次数，空间复杂度为： $O(N)$

// const int MAX_PRICE = 1e5 + 5;
// const int MAX_INVOKING_NUM = 1e5 + 5;
struct stock {
    int price;
    int span;
    stock(int _price, int _span):price(_price), span(_span){};
};

class StockSpanner {
stack<stock> ss;
public:
    StockSpanner() {
        
    }
    
    int next(int price) {
        //栈空和非空的情况合并在一起，代码更为简洁
        int span = 1;
        while(!ss.empty() && ss.top().price <= price) {
            span += ss.top().span;
            ss.pop();
        }
        ss.push(stock(price, span));
        return ss.top().span;
    }
};

/**
 * Your StockSpanner object will be instantiated and called as such:
 * StockSpanner obj = new StockSpanner();
 * int param_1 = obj.next(price);
 */