leetcode 901. 股票价格跨度

最新推荐文章于 2023-10-07 13:06:51 发布

原创最新推荐文章于 2023-10-07 13:06:51 发布 · 684 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈

leetcode 专栏收录该内容

9 篇文章

订阅专栏

本文详细解析了股票跨度查询算法的实现，通过使用递减栈结构，有效地解决了寻找连续日数内股票价格小于或等于今日价格的问题。文章通过具体实例说明了如何维护栈内元素，以快速计算出股票价格的跨度。

编写一个 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);
 */