901. 股票价格跨度 #

原题链接：LeetCode 901. 股票价格跨度

题目描述 #

设计一个算法收集某些股票的每日报价，并返回该股票当日价格的 跨度 。

当日股票价格的 跨度 被定义为股票价格小于或等于今天价格的最大连续日数（从今天开始往回数，包括今天）。

• 例如，如果未来 7 天股票的价格是 [100,80,60,70,60,75,85]，那么股票跨度将是 [1,1,1,2,1,4,6] 。

实现 StockSpanner 类：

• StockSpanner() 初始化类对象。

• int next(int price) 给出今天的股价 price ，返回该股票当日价格的 跨度 。

示例：

输入： ["StockSpanner", "next", "next", "next", "next", "next", "next", "next"][[], [100], [80], [60], [70], [60], [75], [85]] 输出： [null, 1, 1, 1, 2, 1, 4, 6]

• 解释： StockSpanner stockSpanner = new StockSpanner(); stockSpanner.next(100); // 返回 1 stockSpanner.next(80); // 返回 1 stockSpanner.next(60); // 返回 1 stockSpanner.next(70); // 返回 2 stockSpanner.next(60); // 返回 1 stockSpanner.next(75); // 返回 4 ，因为截至今天的最后 4 个股价 (包括今天的股价 75) 都小于或等于今天的股价。 stockSpanner.next(85); // 返回 6

提示：

• 1 <= price <= 105
• 最多调用 next 方法 104 次

难度： Medium

题解代码 #

var StockSpanner = function() {
  this.stack = [] // 价格 单调递减
  this.width = [] // 价格对应的跨度
};

/** 
 * @param {number} price
 * @return {number}
 */
 StockSpanner.prototype.next = function(price) {
  let w = 1
  while (this.stack.length && this.stack[this.stack.length - 1] <= price) {
    this.stack.pop()
    w += this.width.pop()
  }
  this.stack.push(price)
  this.width.push(w)
  return w
};

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