编写一个 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%。
该题力扣
例如
100 20 50 40
当遍历到50的时候可以将20出栈,因为遍历到40的时候50>20,此时20对后面的数没有影响。
如此便可以设计一个单调栈。
栈中记录下标和数值
并且在栈中先插入一个最大值作为天数为 −1天的价格,来保证栈不会为空。
class StockSpanner {
public:
stack<pair<int,int>> stk;//二元数对,记录下标和数值
int idx;//表示当前遍历到第几个元素
StockSpanner() {//初始化栈顶的最大值
stk.emplace(-1,INT_MAX);
idx=-1;
}
int next(int price) {//每次返回与上一个更大元素的下标之差
++idx;
while(price>=stk.top().second) stk.pop();//栈顶元素小于等于price的出栈
int ret=idx-stk.top().first;
stk.emplace(idx,price);
return ret;
}
};
/**
* Your StockSpanner object will be instantiated and called as such:
* StockSpanner* obj = new StockSpanner();
* int param_1 = obj->next(price);
*/
时间复杂度:O(n)
空间复杂度:O(n)