力扣901——股票价格跨度(单调栈)

题目(中等)

编写一个 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%。

思路

小于或等于今天价格的最大连续日数(从今天开始往回数,包括今天)
也就是从今天往前数,直到遇上比今天大的,这个范围内的数的个数。
一开始往平衡树上想了,实现快速插入和查找,后来发现并不是比它小的连续日,只需要从今天开始往前固定起点。
所以用一个单调栈来表示股票,栈内是价格以及往前连续小于它的天数;
对应每一次查询,从栈顶往下找,直到遇到第一个大的就是截至点,这区间内的天数就是答案;
初始化加入{100005, 0},避免每次empty询问;

代码

class StockSpanner {
public:
    stack<pair<int, int>> stock;
    StockSpanner() {
        stock.push({100005, 0});
    }
    
    int next(int price) {
        int num = 1;
        while(price >= stock.top().first) {
            num += stock.top().second;
            stock.pop();
        }
        stock.push({price, num});
        return num;
    }
};

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

你可能感兴趣的:(算法,leetcode,算法,单调栈)