901. 股票价格跨度 - 力扣（LeetCode）

题目

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

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

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

实现 StockSpanner 类：

• StockSpanner() 初始化类对象。
• int next(int price) 给出今天的股价 price ，返回该股票当日价格的 跨度 。

示例：

输入：
["StockSpanner", "next", "next", "next", "next", "next", "next", "next"]
[[], [100], [80], [60], [70], [60], [75], [85]]
输出：
[null, 1, 1, 1, 2, 1, 4, 6]

解释：
StockSpanner stockSpanner = new StockSpanner();
stockSpanner.next(100); // 返回 1
stockSpanner.next(80);  // 返回 1
stockSpanner.next(60);  // 返回 1
stockSpanner.next(70);  // 返回 2
stockSpanner.next(60);  // 返回 1
stockSpanner.next(75);  // 返回 4 ，因为截至今天的最后 4 个股价 (包括今天的股价 75) 都小于或等于今天的股价。
stockSpanner.next(85);  // 返回 6

提示：

• 1 <= price <= 105
• 最多调用 next 方法 104 次

思路

1. 和上一个单调栈的问题也很类似，维护一个保存价格和跨度的栈，当栈为空或者price小于栈顶元素的价格时，入栈并返回1；
2. 反之如果栈顶不为空且price大于栈顶元素的价格时，那么就不断弹栈将栈顶元素的跨度加起来，最后达到终止条件后再加上本身的1作为最终跨度，然后将此跨度入栈并返回跨度。

代码实现

class StockSpanner {
public:
    vector<pair<int, int>> stock_days;
    StockSpanner() {
    }
    int next(int price) {
        if(stock_days.empty()) {
            stock_days.push_back(make_pair(price, 1));
            return 1;
        }
        else {
            int day = 1;
            while(!stock_days.empty() && price >= (stock_days.back()).first) {
                day += (stock_days.back()).second;
                stock_days.pop_back();
            }
            stock_days.push_back(make_pair(price, day));
            return day;
        }
    }
};

/**
 * Your StockSpanner object will be instantiated and called as such:
 * StockSpanner* obj = new StockSpanner();
 * int param_1 = obj->next(price);
 */

复杂度分析

• 时间复杂度：next操作的时间复杂度最坏为O(n)。
• 空间复杂度：O(n)。