LeetCode 每日一题:股票价格跨度

193 阅读1分钟

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

股票价格跨度

原题地址

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

思路分析

以示例中的75为例,来推算一下规律:

  1. 75 当天计算在内得到1;
  2. 前一位60小于75,跟第一步的累加,得到2;
  3. 前一位70小于75,跟第二步的累加,得到3;
  4. 前一位60小于75,跟第三步的累加,得到4;
  5. 前一位80大于75,停止计算,得到75的跨度为4.

根据这个分析,可以得知求75的跨度的时候,只需要得到75前面第一个大于等于75的索引,然后相减即可得到对应的跨度,可以使用单调栈。

  1. 定义一个 stack,以 [price, days] 即价格和当前价格跨度的数组为元素;
  2. next 为计算某个 price 的跨度,因此以上述得到的条件作为终止,累加当前价格的跨度,最后将跨度赋值到对应的价格元素上,然后返回跨度即可。

AC 代码

var StockSpanner = function() {
    this.stack = []; // [[price, days]]
};

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

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

结果:

  • 执行结果: 通过
  • 执行用时:248 ms, 在所有 JavaScript 提交中击败了64.79%的用户
  • 内存消耗:52.9 MB, 在所有 JavaScript 提交中击败了12.67%的用户
  • 通过测试用例:99 / 99

END