- 股票价格跨度
编写一个 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%。
同样利用单调栈,但是由于已经限制好了函数形式,不然解答会更方便的,可以用来存放数组下标,方便计算天数
这里需要另设置一个栈用来记录每一个元素的weight值,存储一个 weight 表示它离上一个价格之间(即最近的一个大于它的价格之间)的天数,最终计算结果的时候方便累加
class StockSpanner {
Stack<Integer> prices, weight;
public StockSpanner() {
prices = new Stack<>();
weight = new Stack<>();//在构造函数中同步化
}
public int next(int price) {
//由于给的不是一个现成的数组,所以需要建立另一个栈,weight
int w= 1;
while(!prices.isEmpty() && prices.peek() <= price){//等于不要忘了
//栈维护的是一个递减的栈,同时相当于去找左边第一个大于price的值
prices.pop();
w += weight.pop();//weight同步出栈,累加其和,每一个在栈中的数都存储着自己的值
}
prices.push(price);
weight.push(w);
return w;
}
}
/**
* Your StockSpanner object will be instantiated and called as such:
* StockSpanner obj = new StockSpanner();
* int param_1 = obj.next(price);
*/