[leetcode 单调栈] 901. 股票价格跨度 M

程序员三木2023-10-08 22:47

设计一个算法收集某些股票的每日报价，并返回该股票当日价格的跨度。

当日股票价格的跨度被定义为股票价格小于或等于今天价格的最大连续日数（从今天开始往回数，包括今天）。

例如，如果未来 7 天股票的价格是 $100,80,60,70,60,75,85$ ，那么股票跨度将是 $1,1,1,2,1,4,6$ 。

实现 StockSpanner 类：

StockSpanner() 初始化类对象。

int next(int price) 给出今天的股价 price ，返回该股票当日价格的跨度。

示例：

输入：

"StockSpanner", "next", "next", "next", "next", "next", "next", "next"

\], \[100\], \[80\], \[60\], \[70\], \[60\], \[75\], \[85

输出：

null, 1, 1, 1, 2, 1, 4, 6

解释：

StockSpanner stockSpanner = new StockSpanner();

stockSpanner.next(100); // 返回 1

stockSpanner.next(80); // 返回 1

stockSpanner.next(60); // 返回 1

stockSpanner.next(70); // 返回 2

stockSpanner.next(60); // 返回 1

stockSpanner.next(75); // 返回 4 ，因为截至今天的最后 4 个股价 (包括今天的股价 75) 都小于或等于今天的股价。

stockSpanner.next(85); // 返回 6

提示：

1 <= price <= 105

最多调用 next 方法 104 次

暴力即美学

java 复制代码

class StockSpanner {

    
    private final List<Integer> arr;

    public StockSpanner() {
        arr = new ArrayList<>();
    }

    public int next(int price) {
        arr.add(price);
        int ans = 1;
        for(int i=arr.size()-2; i>=0; i--) {
            if(arr.get(i) <= price) {
                ++ans;
            } else {
                break;
            }
        }
        return ans;
    }

}

/**
 * Your StockSpanner object will be instantiated and called as such:
 * StockSpanner obj = new StockSpanner();
 * int param_1 = obj.next(price);
 */

单调栈

java 复制代码

class StockSpanner {
    Deque<int[]> stack;
    int idx;

    public StockSpanner() {
        stack = new ArrayDeque<int[]>();
        stack.push(new int[]{-1, Integer.MAX_VALUE});
        idx = -1;
    }

    public int next(int price) {
        idx++;
        while (price >= stack.peek()[1]) {
            stack.pop();
        }
        int ret = idx - stack.peek()[0];
        stack.push(new int[]{idx, price});
        return ret;
    }
}