中等

问题描述

编写一个 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%。

单调栈

建立一个栈，栈内以二元组 $(price, idx)$ 形式维护比当前元素 $price$ 大的元素及其下标。 每次调用 $next$ 操作时，从栈顶开始处理，将所有满足不大于 $price$ 的元素进行出栈，从而找到当前元素 $price$ 左边最近一个比其大的位置，计算当前坐标与该元素坐标距离即可。

var StockSpanner = function () {
  this.i = 1
  this.stack = []
}

/**
 * @param {number} price
 * @return {number}
 */
StockSpanner.prototype.next = function (price) {
  while (this.stack.length && this.stack[this.stack.length - 1][0] <= price) {
    this.stack.pop()
  }
  let ans = this.i
  if (this.stack.length) {
    ans = this.i - this.stack[this.stack.length - 1][1]
  }
  this.stack.push([price, this.i++])
  return ans
}

/**
 * Your StockSpanner object will be instantiated and called as such:
 * var obj = new StockSpanner()
 * var param_1 = obj.next(price)
 */

• 时间复杂度：$next$ 操作的均摊复杂度为 $O(1)$
• 空间复杂度：$O(n)$