单调栈作为一种数据结构在求解类递增、递减方面的题目中有较为广泛的应用,在以往的leetcode中所见到的相关单调栈的题目均为单一元素,今天刷到901题目时,想到了将数组元素作为单调栈中元素的方法进行求解。
题目链接及描述
901. 股票价格跨度 - 力扣(LeetCode)
题目分析
做这到题目首先想到的是使用一个数组将其元素依次存起来,随后将其转换为一个数组相关的题目,但由于数组需要实现开辟好空间,不太好确定,考虑使用List来实现,当调用next方法就将其存储到List集合中。此时List集合中的元素类型为一个一维数组,并且数组大小为2,第一个元素为price,第二个元素为price对应的价格跨度。
将示例代码构造为阶梯图如下所示:
最左边元素为Integer.MAX_VALUE,添加此元素,可以减少栈的非空判断。减少代码复杂度。
根据题目:当日股票价格的 跨度 被定义为股票价格小于或等于今天价格的最大连续日数。故单调栈中存储的元素应为递减的,每当调用依次next方法,需要不断循环弹出当前栈顶小于当前价格price的元素,最后当前价格price对应的下标 - 栈顶元素对应的下标即为所求结果。当遍历到75时当前栈如下所示:
遍历到元素75,此时栈中的情况:上图中下标2和下标4对应的蓝色柱形条表示已经出栈的元素。
循环遍历当前栈顶元素与元素75进行比较,最终下标3对应的元素70出栈,则75对应的结果为5 - 1 = 4。
代码编写:
class StockSpanner {
public Deque<int[]> mst;
public int idx;
public StockSpanner() {
this.mst = new ArrayDeque<>();
this.idx = 0;
mst.addLast(new int[]{Integer.MAX_VALUE, -1});
}
public int next(int price) {
while(!mst.isEmpty() && price >= mst.peekLast()[0]){
mst.pollLast();
}
int ans = idx - mst.peekLast()[1];
mst.addLast(new int[]{price, idx++});
return ans;
}
}