901. 股票价格跨度(难度:中等)
2024-04-09 18:20:48  阅读数 360

题目链接:https://leetcode.cn/problems/online-stock-span/

题目描述:

编写一个 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. 每个测试用例最多可以调用 10000StockSpanner.next
  3. 在所有测试用例中,最多调用 150000StockSpanner.next
  4. 此问题的总时间限制减少了 50%。

解法:栈

根据题目描述,每次需要找到当前元素之前的小于等于自己的连续的元素个数。

那其实每次只需要关注,前面有多少个小于等于当前元素的个数,当元素小于遍历到的元素时,将不再关注前面的元素。

我们可以利用一个栈结构stack,里面每一个元素存放一个数组temp[2],来存储前面的元素大小temp[0]和小于等于该元素的个数temp[1]。

每次next时,就遍历栈,和栈首元素temp = stack.peekFirst()进行比较:

  • 当price <= temp[0] 时,将栈首元素出栈,并将元素个数加上栈首元素中的个数,即 result += temp[1]。
  • 当price > temp[0] 时,则退出栈遍历。

最后将该元素的大小和小于等于该元素的个数入栈。

代码:

class StockSpanner {
    Deque<int[]> queue;

    public StockSpanner() {
        queue = new ArrayDeque<>();
    }

    public int next(int price) {
        int result = 1;
        int[] temp = {price, result};
        while (!queue.isEmpty()) {
            if (queue.peekFirst()[0] <= price) {
                result += queue.pollFirst()[1];
            } else {
                break;
            }
        }
        temp[1] = result;
        queue.offerFirst(temp);
        return result;
    }
}