编写一个
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]
看到题目的第一直觉就是栈,一开始抓不住规律。
其实我们每次next找的都是在新元素左边的比新元素大的最后一个数,因此维护一个递减的栈,栈里存数字和数字的下标,每次和栈顶比较, 如果小于的话,那么以这个下标为准,后面的数字都比新元素小。而如果新元素大于栈顶数字,就一直出栈,直到栈顶元素大于新元素。
class StockSpanner {
private ArrayList<Integer> list;
private Stack<int[]> num;
public StockSpanner() {
list = new ArrayList<Integer>();
num = new Stack<int[]>();
}
public int next(int price) {
while(!num.isEmpty() && num.peek()[0] <= price){
num.pop();
}
int res = 1;
if(num.isEmpty()){
res = list.size() + 1;
}
else
res = list.size() - num.peek()[1];
list.add(price);
num.push(new int[]{price,list.size() - 1});
return res;
}
}
/**
* Your StockSpanner object will be instantiated and called as such:
* StockSpanner obj = new StockSpanner();
* int param_1 = obj.next(price);
*/