单调栈的实现-单调递减栈和单调递增栈(java)

什么是单调栈

单调栈是一种常用的数据结构和算法，用于解决一些具有单调性质的问题，例如找到下一个更大的元素或下一个更小的元素等。

单调栈的基本思想是维护一个递增或递减的栈，使得每次查询时可以在 O(1) 的时间复杂度内找到下一个更大或更小的元素。具体地，假设我们要找到一个数组中下一个更大的元素，我们可以维护一个递减的栈，将数组中的元素按照降序的方式压入栈中。这样，对于每个元素，我们只需要检查它的下一个元素是否比栈顶元素大，如果是，则弹出栈顶元素，直到找到下一个更大的元素。

单调递增栈:

①在一个队列中针对每一个元素从它右边寻找第一个比它小的元素
②在一个队列中针对每一个元素从它左边寻找第一个比它小的元素

单调递减栈

①在一个队列中针对每一个元素从它右边寻找第一个比它大的元素
②在一个队列中针对每一个元素从它左边寻找第一个比它大的元素

单调递增栈

本文重点讲述下,单调递增栈,因为递增和递减逻辑上是一样的,所以只需要搞懂一个就行了.
先看个示例,明白要的是个什么样的数据结构:
原始 数组:arr = [ 3, 1, 2, 3]
生成的单调栈:
[
0 : [-1, 1]
1 : [-1, -1]
2 : [ 1, -1]
3 : [ 2, -1]
]
解释:
单调栈的数据结构维护的是原数组的下标,每个小数组两个位置,第一个位置维护的是:左边第一个比它小的元素
第二个位置维护的是:右边第一个比它大的元素
0:[-1,1] 代表原数组第一个位置3 左边没有比他小的,用-1占位,右边第一个比他小的是1位置的1.

我们怎么实现这个过程呢,把数组的元素生成一个单调栈的二维数组.
从代码里看流程:

代码演示

   /**
     * 有重复值的情况下 实现单调栈
     * 队列中每个位置维护一个集合.集合中存放下标值,相等的值,加到集合的后面.
     * @param arr
     * @return
     */
    public static int[][] getNearLess(int[] arr) {
        int[][] ans = new int[arr.length][2];
        //栈里保存数组的下标.
        Stack<List<Integer>> stack = new Stack<>();
        for (int i = 0; i < arr.length;i++){
            //当前队列不为空,且新加入的数字比之前的小时,单调栈的结构需要调账
            while (!stack.isEmpty() && arr[stack.peek().get(0)] > arr[i]){
                List<Integer> pop = stack.pop();
                //找到左边第一个出现比其小的值.
                int lessLeft = stack.isEmpty() ? -1 : stack.peek().get(stack.peek().size() - 1);
                for (int pi : pop){
                    ans[pi][0] = lessLeft;
                    ans[pi][1] = i;
                }
            }
            //如果相等时,队列不为空,加到数组中
            if (!stack.isEmpty() && arr[stack.peek().get(0)] == arr[i]){
                stack.peek().add(Integer.valueOf(i));
            }else {
                //小于的时候,或者队列为null时
                ArrayList<Integer> list = new ArrayList<>();
                list.add(i);
                stack.push(list);
            }

        }
        //剩余最后的
        while (!stack.isEmpty()){
            List<Integer> pops = stack.pop();
            int lessLeft = stack.isEmpty() ? -1 : stack.peek().get(stack.peek().size() - 1);
            for (int pi : pops){
                ans[pi][0] = lessLeft;
                ans[pi][1] = -1;
            }
        }
        return ans;
    }

往期算法经典

leetcode474. 一和零

leetcode97. 交错字符串

leetcode464. 我能赢吗

leetcode957. N 天后的牢房

leetcode526. 优美的排列

leetcode667. 优美的排列 II

leetcode902. 最大为 N 的数字组合