【leetcode】901. Online Stock Span

转载于 2018-09-17 17:50:00 发布 · 86 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/seyjs/p/9613320.html

文章标签：

#数据结构与算法

本文介绍了一个股票跨度问题的解决方法，其核心思路与LeetCode 84题最大矩形相似，旨在寻找当前元素前最近的大于自身的元素。通过维护两个列表实现高效查找，最终返回每个价格对应的跨度值。

题目如下：

解题思路：和【leetcode】84. Largest Rectangle in Histogram的核心是一样的，都是要找出当前元素之前第一个大于自己的元素。

代码如下：

class StockSpanner(object):
    def __init__(self):
        self.cl = []
        self.vl = []

    def next(self, price):
        """
        :type price: int
        :rtype: int
        """
        count = 1
        if len(self.vl) == 0:
            self.vl.append(price)
            self.cl.append(1)
        else:
            startInx = len(self.vl) - 1
            while startInx >= 0 and price >= self.vl[startInx]:
                count += self.cl[startInx]
                startInx -= self.cl[startInx]
            self.vl.append(price)
            self.cl.append(count)
        return count