地址:https://leetcode-cn.com/problems/online-stock-span/
知识点:
- 找左边连续的小于等于自己的元素,单调栈就是专门解决这种问题的,因此除了暴力法就是这种方法;
- 单调栈问题一般都记录索引;
- 这里数据是动态的,很方便的一点是,可以使用哨兵,这样就不用考虑特殊情况了。
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;
public class StockSpanner {
private List<Integer> stock;
/**
* Java 官方推荐使用 Deque,只用和栈相关的接口
*/
private Deque<Integer> indexes;
private int index;
public StockSpanner() {
// 哨兵,这个元素永远不会出栈
stock = new ArrayList<>();
stock.add(10_0000 + 1);
indexes = new ArrayDeque<>();
indexes.addLast(0);
index = 0;
}
public int next(int price) {
index++;
// 特别注意:不要用 indexes.peek(),这个方法等价于 peekFirst()
while (!indexes.isEmpty() && stock.get(indexes.peekLast()) <= price) {
indexes.removeLast();
}
// 因为 indexes 后面会更改,因此这里先把结果暂存一下
int res = index - indexes.peekLast() ;
stock.add(price);
indexes.addLast(index);
return res;
}
}
李威威
发布了435 篇原创文章 · 获赞 325 · 访问量 123万+
关注