leetcode 901 股票价格跨度

用栈来求解上一个最大的元素

class StockSpanner {
    Deque 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;
    }
}

你可能感兴趣的:(leetcode,leetcode,算法,java)