用栈来求解上一个最大的元素
class StockSpanner {
Deque<int[]> stack;
int index = -1;
public StockSpanner() {
//求下一个更大元素
//因为求的是前面连续几天小于等于当前价格,用栈来存储
//栈每一个元素都是数组,第一个元素是当前是第几天,第二个元素是当前的价格
//遇到比自己价格小的就从栈弹出,直到遇到大的,用当前天数减去大的天数就是有几个连续天数
stack = new ArrayDeque<>();
//先放入最大值,保证如果有价格都大于前面减去-1
stack.add(new int[]{index,Integer.MAX_VALUE});
}
public int next(int price) {
//当前天数
index++;
//遇到小的就弹出
while (price >= stack.getLast()[1])
stack.removeLast();
//此时栈顶是大的
int res = index - stack.getLast()[0];
stack.add(new int[]{index,price});
return res;
}
}