Keshawn_lu's Blog

Leetcode 901. 股票价格跨度

单调栈
字数统计: 510阅读时长: 2 min
2022/10/21 Share

题目简介：

编写一个 StockSpanner 类，它收集某些股票的每日报价，并返回该股票当日价格的跨度。

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

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

示例：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
输入：["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%。

思路：

利用单调栈，若栈顶元素的price小于当前的price，则不断出栈。

全部出栈结束后，当前元素的day - 栈顶元素的day即为最大连续日数

tip:

  • 每次入栈时，将下标(day)和price一起入栈。
  • 初始化时，压入的元素day=0, price=INT_MAX。

代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
class StockSpanner {
public:

    stack<pair<int, int>> stk;
    int day;

    StockSpanner() {

        day = 0;
        stk.push(make_pair(0, INT_MAX));
    }
    
    int next(int price) {

        day++;
        while(!stk.empty() && price >= stk.top().second){

            stk.pop();
        }

        int res = day - stk.top().first;
        stk.push(make_pair(day, price));

        return res;
    }
};

/**
 * Your StockSpanner object will be instantiated and called as such:
 * StockSpanner* obj = new StockSpanner();
 * int param_1 = obj->next(price);
 */
CATALOG
  1. 1. 题目简介：
  2. 2. 思路：
  3. 3. 代码如下：
Total : 424
2023
2022
2021
2020
2019
动态规划 树 深度优先搜索 哈希表 数组 深度优先遍历 单调栈 字符串 双指针 栈 贪心 图 模拟 广度优先搜索 回溯 滑动窗口 二分查找 广度优先遍历 记忆化搜索 数学 链表 前缀和 双向链表 优先队列 拓扑排序 KMP 堆 字典树 递归 并查集 矩阵 位运算 双端队列 吴恩达团队NLP 吴恩达深度学习 线性表
杂谈 算法相关 PAT 2019推理小说阅读计划 NLP 深度学习 大三课程学习 大四课程学习 浙工商OJ 牛客网