Leetcode 901. 股票价格跨度

最新推荐文章于 2024-02-18 23:30:27 发布

Richest_li

最新推荐文章于 2024-02-18 23:30:27 发布

阅读量89

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/li_yizhixiaowukong/article/details/127444874

版权

Leetcode 专栏收录该内容

72 篇文章 1 订阅

订阅专栏

编写一个 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.最开始想着，把每次的价格都放一个列表中。每次从后往前遍历列表，当有一个比 price 小的，ans+=1；当碰到比 price 大的，退出循环。（因为它要最大连续日数）

class StockSpanner:

    def __init__(self):
        self.stacklist = list()

    def next(self, price: int) -> int:
        ans = 1
        listlength = len(self.stacklist)
        for i in reversed(range(listlength)):
            if self.stacklist[i] <= price:
                ans += 1
            else:
                break
        self.stacklist.append(price)
        return ans

又直接超时。没过的测试案例老长了，刚把测试案例粘过来，提示读完需要4个小时。

改进的想法还是从后往前遍历，只不过之前 for 循环是一步一步遍历。
改成 while 循环，多引一个 last 列表用来放每个 price 对应的 ans 值。
当 stacklist[j] 小于等于 price 时，如果 last 列表不为空，即 j 不为 0 时：
ans += （ j 对应的 ans 值），j 前移（ j 对应的 ans 值）。
当 stacklist[j] 大于 price 时，退出循环：

class StockSpanner:

    def __init__(self):
        self.stacklist = list()
        self.last = list()

    def next(self, price: int) -> int:
        ans = 1
        listlength = len(self.stacklist)
        j = listlength - 1
        while j >= 0:
            if self.stacklist[j] <= price:
                if self.last :
                    ans += self.last[j]
                    j -= self.last[j]
                else:
                    ans += 1
                    j -= 1
            else:
                break
        self.stacklist.append(price)
        self.last.append(ans) 
        return ans