每日一题 901. 股票价格跨度

95 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第23天,点击查看活动详情

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) 小于或等于今天的价格。

提示:

  1. 调用 StockSpanner.next(int price) 时,将有 1 <= price <= 10^5
  2. 每个测试用例最多可以调用  10000 次 StockSpanner.next
  3. 在所有测试用例中,最多调用 150000 次 StockSpanner.next
  4. 此问题的总时间限制减少了 50%。

解题思路

今天(today)往前走:

  • 如果前面的一天(before)股票价值大于今天
    • 总共的天数就是:
      • today - before:其实就等于两天的下标相减
  • 如果没有,没什么好说的就是:1

那么我们就可以使用栈,栈中存放[某天的下标, 当天的股票价值],我们需要注意的一点是,我们需要先初始化这个栈,保证栈不为空

var StockSpanner = function () {
	this.days = -1;
	this.list = [[this.days, Number.MAX_VALUE]];
};

/**
 * @param {number} price
 * @return {number}
 */
StockSpanner.prototype.next = function (price) {
    // 每次输入 说明事件就过去了一天
	this.days++;
    // 输入的价格和之前的进行比较
	while (price >= this.list[this.list.length - 1][1]){
        // 小于今天价格的直接出栈
        this.list.pop();
    }
    // 今天对比之后的值
    let value = this.days - this.list[this.list.length - 1][0]
    // 然后为栈新增今天的数据
	this.list.push([this.days, price]);
	return value;
};

image.png