单调栈题目：股票价格跨度

题目

标题和出处

标题：股票价格跨度

出处：901. 股票价格跨度

难度

6 级

题目描述

要求

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

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

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

实现 $\text{StockSpanner}$ 类：

• $\text{StockSpanner()}$ 初始化类的对象。
• $\text{int next(int price)}$ 给定今天的股票价格是 $\text{price}$，返回今天股票价格的跨度。

示例

示例 1：

输入：
$\text{["StockSpanner", "next", "next", "next", "next", "next", "next", "next"]}$
$\text{[[], [100], [80], [60], [70], [60], [75], [85]]}$
输出：
$\text{[null, 1, 1, 1, 2, 1, 4, 6]}$
解释：
$\text{StockSpanner stockSpanner = new StockSpanner();}$
$\text{stockSpanner.next(100);}$ // 返回 $\text{1}$
$\text{stockSpanner.next(80);}$ // 返回 $\text{1}$
$\text{stockSpanner.next(60);}$ // 返回 $\text{1}$
$\text{stockSpanner.next(70);}$ // 返回 $\text{2}$
$\text{stockSpanner.next(60);}$ // 返回 $\text{1}$
$\text{stockSpanner.next(75);}$ // 返回 $\text{4}$，因为截至今天的最后 $\text{4}$ 个价格（包括今天的价格 $\text{75}$）小于或等于今天的价格
$\text{stockSpanner.next(85);}$ // 返回 $\text{6}$

数据范围

• $\text{1}\le \text{price}\le {\text{10}}^{\text{5}}$
• 至多调用 $\text{next}$ 方法 ${\text{10}}^{\text{4}}$ 次

解法

思路和算法

题目要求计算股票当日价格的跨度，即从当前日期开始往回数，股票价格小于或等于当日价格的最大连续日数。为了得到股票当日价格的跨度，需要找到历史数据中的股票价格大于当日价格的最大日期。假设当前日期是 $i$，历史数据中的股票价格大于日期 $i$ 价格的最大日期是 $j$，其中 $j<i$，则从日期 $j+1$ 到日期 $i$ 的连续 $i-j$ 天都满足股票价格小于或等于日期 $i$ 的价格，日期 $i$ 的股票价格跨度即为 $i-j$。

为了计算股票价格跨度，可以使用单调栈。由于需要比较股票价格和计算跨度，因此单调栈需要同时存储日期和股票价格，满足从栈底到栈顶的股票价格单调递减。单调栈内的元素格式是 $[\text{index},\text{price}]$，其中 $\text{index}$ 表示日期，$\text{price}$ 表示该日期的股票价格，$\text{index}$ 的初始值是 $0$。

对于 $\text{next}$ 方法，进行如下操作：

1. 如果栈不为空且栈顶元素的股票价格小于或等于当日股票价格，则将栈顶元素出栈，重复该操作直到栈为空或者栈顶元素的股票价格大于当日股票价格；

2. 计算历史数据中的股票价格大于当日股票价格的最大日期 $\text{prev}$，如果栈为空则 $\text{prev}$ 为 $-1$，如果栈不为空则 $\text{prev}$ 为栈顶元素的日期，当日股票价格跨度是 $\text{span}=\text{index}-\text{prev}$；

3. 将 $[\text{index},\text{price}]$ 入栈，然后将 $\text{index}$ 加 $1$；

4. 返回 $\text{span}$。

证明

由于单调栈满足从栈底到栈顶的股票价格单调递减，因此当 $[\text{index},\text{price}]$ 入栈时，单调栈内的所有元素的股票价格都大于 $\text{price}$。考虑 $\text{price}$ 入栈之前的栈顶日期 $\text{prev}$（如果栈为空则 $\text{prev}=-1$），$\text{prev}$ 一定是历史数据中股票价格大于 $\text{price}$ 的最大日期。理由如下：

• 如果 $\text{prev}=-1$，则历史数据中的股票价格一定都小于或等于 $\text{price}$，否则大于 $\text{price}$ 的股票价格一定会在单调栈内；

• 如果 $\text{prev}\ge 0$，假设存在日期 $\text{mid}$ 满足 $\text{prev}<\text{mid}<\text{index}$ 且日期 $\text{mid}$ 的股票价格大于当日股票价格 $\text{price}$，则无论日期 $\text{mid}$ 的股票价格是否小于日期 $\text{prev}$ 的股票价格，日期 $\text{mid}$ 一定会入栈，此时栈顶日期是 $\text{mid}$ 或者大于 $\text{mid}$ 的某个日期，出现矛盾，因此 $\text{prev}$ 一定是历史数据中股票价格大于 $\text{price}$ 的最大日期。

由于栈顶日期 $\text{prev}$ 是历史数据中股票价格大于 $\text{price}$ 的最大日期，因此从日期 $\text{prev}+1$ 到 $\text{index}$ 的股票价格都小于或等于当日股票价格，当日股票价格的跨度即为 $\text{index}-\text{prev}$。

代码

class StockSpanner {
    Deque<int[]> stack;
    int index;

    public StockSpanner() {
        stack = new ArrayDeque<int[]>();
        index = 0;
    }
    
    public int next(int price) {
        while (!stack.isEmpty() && stack.peek()[1] <= price) {
            stack.pop();
        }
        int prev = stack.isEmpty() ? -1 : stack.peek()[0];
        int span = index - prev;
        stack.push(new int[]{index, price});
        index++;
        return span;
    }
}

复杂度分析

• 时间复杂度：构造方法的时间复杂度是 $O(1)$，方法 $\text{next}$ 的均摊时间复杂度是 $O(1)$。假设总天数是 $n$，则共有 $n$ 个日期和股票价格，每个元素（包含日期和对应的股票价格）最多入栈和出栈各一次，因此均摊时间复杂度是 $O(1)$。

• 空间复杂度：$O(n)$，其中 $n$ 是总天数。空间复杂度主要取决于栈空间，栈内元素个数不会超过 $n$。