LeetCode每日一题——901. 股票价格跨度

LeetCode每日一题系列

题目：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%。

思路

单调栈，题中意思等价于寻找大于某元素上一个值，然后求坐标差值就是答案了。
这里由于题意，我们需要存储输入元素以及输入元素的下标，输入元素直接存入数组，下标存入单调栈，单调栈自栈顶到栈底递增，栈顶元素即为大于某一元素的上一个值的下标。
这里求差值时还需判断单调栈是否为空，如果为空说明某元素大于前方所有元素，这里返回下标+1即可

题解

class StockSpanner:

    def __init__(self):
    	# 单调栈
        self.s = []
        # 输入元素数组
        self.num = []
        # 下标
        self.index = -1

    def next(self, price: int) -> int:
        self.index += 1
        # 从单调栈弹出元素
        while self.s and self.num[self.s[-1]] <= price:
            self.s.pop()
      	# 求答案
        ans = self.index - self.s[-1] if self.s else self.index + 1
        # 更新单调栈以及输入元素数组
        self.s.append(self.index)
        self.num.append(price)
        return ans