LeetCode 901 Online Stock Span (Python)

Posted by 小明MaxMing on May 19, 2020

题目

Write a class StockSpanner which collects daily price quotes for some stock, and returns the span of that stock’s price for the current day.

The span of the stock’s price today is defined as the maximum number of consecutive days (starting from today and going backwards) for which the price of the stock was less than or equal to today’s price.

For example, if the price of a stock over the next 7 days were [100, 80, 60, 70, 60, 75, 85], then the stock spans would be [1, 1, 1, 2, 1, 4, 6].

解题思路

维护一个严格递降的栈,在入栈的时候,也把当前的span加进去,这样在pop的时候,就把被pop的span加到新的数中

代码

class StockSpanner:

    def __init__(self):
        self.stack = [(0, float('inf'))]
        
    def next(self, price: int) -> int:
        res = 1
        while price >= self.stack[-1][1]:
            res += self.stack.pop()[0]
        self.stack.append((res, price))
        return res

视频讲解 YouTube<--欢迎点击订阅

视频讲解 bilibili<--欢迎点击订阅