leetcode刷题记录-901. 股票价格跨度

152 阅读1分钟

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

前言

今天的题目为中等,简单题,维护一个数组,用来保存题目要求的数字,按照题目的要求进行筛选即可。

每日一题

今天的题目是 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%。

题解

单调栈

按照题目的要求,每次调用next的时候,要返回今天股票价格的跨度 也就是 定义为股票价格小于或等于今天价格的最大连续日数(从今天开始往回数,包括今天)。例子中就有稍微提到了,next(75) 的话返回的是4,也就是 两个60 一个70 包括今天,所以是四天。

那么我们利用一个数组来保存当前的状态,在每次调用next的时候,需要去判断前面的数组中有多少个是满足上面的要求的,然后返回今日加上之前满足条件的天数之和。

在最开始的初始化的时候,可以用一个max的number来作为第一项,这样不会影响到之后的计算,因为是最大的数,之后的数不可能比它大,可以省去判空。

class StockSpanner {
    arr: any[] = []
    index: number = -1
    constructor() {
        this.arr.push({index: -1, price: Number.MAX_VALUE})
    }

    next(price: number): number {
        this.index++;
        while (price >= this.arr[this.arr.length - 1].price) {
            this.arr.pop();
        }
        let ret = this.index - this.arr[this.arr.length - 1].index;
        this.arr.push({index: this.index, price: price});
        return ret;
    }
}

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

image.png