编写一个 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%。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/online-stock-span
解题思路:首先题目是要求我们从今天开始往之前去找价格不比我们高的天数(包括今天),很直观的想法是记录下所有的数据然后从列表后面开始遍历,这样效率会很低。因为我们会输出每一个价格和在它之前比它低的天数,已经包含了历史信息再去遍历不值得。仔细观察它的数据结构,我们只需要在栈中记录下一个价格依次降低的历史数据和它们相对应的跨度,这样就可以保留所有的历史信息。AC代码如下:
class StockSpanner:
def __init__(self):
self.stack = []
def next(self, price: int) -> int:
value = 1
while self.stack:
if price >= self.stack[len(self.stack) - 1][1]:
vs = self.stack.pop()[0]
value += vs
else:
self.stack.append((value, price))
return value
if not self.stack:
self.stack.append((value, price))
return value