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%。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/online-stock-span
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

总思路：当前股价往前搜索，遇到往日股价更大的则停止

设计两个全局变量stack，用于存储股价 & 跨度
维护一个价格的单调栈
若当前股价 > 栈顶股价，则当前股价的跨度应加上栈顶股价的跨度

class StockSpanner {

    var priceStack: [Int]
    var spanStack: [Int]
    
    init() {
        priceStack = [Int]()
        spanStack = [Int]()
    }
    
    func next(_ price: Int) -> Int {
        var span = 1
        while let lastPrice = priceStack.last, let lastSpan = spanStack.last, price >= lastPrice {
            span += lastSpan
            spanStack.removeLast()
            priceStack.removeLast()
        }
        
        priceStack.append(price)
        spanStack.append(span)
        return span
    }
}

/**
 * Your StockSpanner object will be instantiated and called as such:
 * let obj = StockSpanner()
 * let ret_1: Int = obj.next(price)
 */

其他思路

https://www.bilibili.com/video/BV1kW411k77K?from=search&seid=15780837099295797199