传送门:https://leetcode-cn.com/contest/weekly-contest-101/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) 小于或等于今天的价格。
提示:
- 调用
StockSpanner.next(int price)
时,将有1 <= price <= 10^5
。 - 每个测试用例最多可以调用
10000
次StockSpanner.next
。 - 在所有测试用例中,最多调用
150000
次StockSpanner.next
。 - 此问题的总时间限制减少了 50%。
分析:维护一个单调(递减)栈,栈中元素为股票跨度中的最大元素
时间复杂度: 为调用次数, 空间复杂度为:
// const int MAX_PRICE = 1e5 + 5;
// const int MAX_INVOKING_NUM = 1e5 + 5;
struct stock {
int price;
int span;
stock(int _price, int _span):price(_price), span(_span){};
};
class StockSpanner {
stack<stock> ss;
public:
StockSpanner() {
}
int next(int price) {
//栈空和非空的情况合并在一起,代码更为简洁
int span = 1;
while(!ss.empty() && ss.top().price <= price) {
span += ss.top().span;
ss.pop();
}
ss.push(stock(price, span));
return ss.top().span;
}
};
/**
* Your StockSpanner object will be instantiated and called as such:
* StockSpanner obj = new StockSpanner();
* int param_1 = obj.next(price);
*/