持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第23天，点击查看活动详情

力扣——901. 股票价格跨度

901. 股票价格跨度 - 力扣（LeetCode）

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

问题解析

这题就是要我们求：对于第i个数，从第i个位置到第1个位置，有连续多少个数小于等于它。

也就是说我们要知道对于当前数，之前比他大的最近的数在哪里，我们可以用单调栈来实现他。栈存储元素为数对，first是那个元素的大小，second是那个元素的位置。

对于插入的数：

• 如果栈为空或栈顶的first大于他，我们将该元素入栈。
• 如果栈顶的first小于等于他，就一直出栈，直到栈顶元素大于当前元素或者栈为空为止。再将该元素入栈。

答案就为当前元素的位置到栈顶元素（当前元素入栈前的栈顶元素）的位置+1

AC代码

class StockSpanner {
public:
    stack<pair<int,int>>sta;
    int cnt=1;
    StockSpanner() {
​
    }
    
    int next(int price) {
        if(sta.empty()||sta.top().first>price)
        {
            sta.push({price,cnt++});
            return 1;
        }
        while(!sta.empty()&&sta.top().first<=price)
        {
            sta.pop();
        }
        int x=cnt;
        if(!sta.empty())x-=sta.top().second;
        sta.push({price,cnt++});
        return x;
    }
};
​
/**
 * Your StockSpanner object will be instantiated and called as such:
 * StockSpanner* obj = new StockSpanner();
 * int param_1 = obj->next(price);
 */