力扣:单调栈算法思路题

单调栈分为单调递增栈和单调递减栈,通过使用单调栈我们可以访问到最近一个比它大(小)的元素。

 单调递增栈:单调递增栈就是从栈底到栈顶数据是依次递增,通常是寻找某方向第一个比它小的元素。
 单调递减栈:单调递减栈就是从栈底到栈顶数据是依次递减,通常是寻找某方向第一个比它大的元素。

适用场景
什么情况适合用单调栈来解决实际问题呢?
通常是在数组中需要通过比较前后元素的大小关系来找最近的比它大(小)的元素问题时,可以使用单调栈进行求解。

场景示例
1:寻找左边第一个小于它的数

/**
     * 寻找左边第一个小于它的数
     * 单调递增栈:单调递增栈就是从栈底到栈顶数据是依次递增,通常是寻找某方向第一个比它小的元素
     * 单调递减栈:单调递减栈就是从栈底到栈顶数据是依次递减,通常是寻找某方向第一个比它大的元素
     *
     * 题目描述: 给定一个长度为 n ≤ 10 ^5 的数组 a,输出每个数左边第一个比它小的数,如果不存在则输出 − 1。
     *
     * 在指针 i 从左往右遍历的过程中,我们可以用一个栈来保存 i 左边的所有元素(不包括i指向的元素)
     * ,下标越大的元素越接近栈顶,下标越小的元素越接近栈底。
     * 每次我们访问栈顶,只要栈顶元素大于等于 a [ i ],我们就将栈顶元素弹出,直至栈顶元素小于 a [ i ] ,
     * 此时输出栈顶元素并将 a [ i ] 压入栈中。 由于栈中保存了 i 左边的所有元素,所以只要有答案,则答案一定在栈中。
     * 由于每个元素一定会被压入一次且至多弹出一次,因此操作次数至多是2n,故总时间复杂度为O(n)
     * @param array
     * @return
     */
    public static int[] findFirstLeftLower(int[] array){
        Deque<Integer> linkList = new LinkedList<>();
        int[] ans = new int[array.length];
        for (int i = 0; i < array.length; i++) {
            // 如果栈不为空且当前数小于等于栈顶元素,则将栈顶出栈,并通过linkList.push(array[i])将当前元素入栈
            while(!linkList.isEmpty() && array[i] <= linkList.peek()){
            // 如果是求右边第一个大于它的数,只需要替换成  array[i] >= linkList.peek()
                linkList.poll();
            }
            if(!linkList.isEmpty()){
                // 由于栈顶元素存放第一个比当前元素小的数,则取出并给结果数组赋值
                ans[i] = linkList.peek();
            }else{
                ans[i] = -1;
            }
            linkList.push(array[i]);
        }

       /* for (int i = 0; i < ans.length; i++) {
            System.out.print(ans[i]+" ");
        }*/
        return ans;
    }

2:寻找左边第一个小于它的数的下标

 /**
     * 寻找左边第一个小于它的数的下标
     * 单调递增栈:单调递增栈就是从栈底到栈顶数据是依次递增,通常是寻找某方向第一个比它小的元素
     * 单调递减栈:单调递减栈就是从栈底到栈顶数据是依次递减,通常是寻找某方向第一个比它大的元素
     *
     * 题目描述: 给定一个长度为 n ≤ 10 ^5 的数组 a,输出每个数左边第一个比它小的数的下标,如果不存在则输出 − 1。
     * 我们只需要注意几个点,在当前条件下,咱们栈中存的是下标,而不是值,
     * 所以需要修改两个地方:
     * a[linkList.peek()] 而不是linkList.peek(),
     * 不再是a[i],而是存储对应的下标i
     * @param array
     * @return
     */
    public static int[] findFirstLeftLowerPosition(int[] array){
        Deque<Integer> linkList = new LinkedList<>();
        int[] ans = new int[array.length];
        for (int i = 0; i < array.length; i++) {
            // 如果栈不为空且当前数小于等于栈顶元素,则将栈顶出栈,并通过linkList.push(array[i])将当前元素入栈
            while(!linkList.isEmpty() && array[i] <= array[linkList.peek()]){
                // 如果是求右边第一个大于它的数的下标,只需要替换成  array[i] >= linkList.peek()
                linkList.poll();
            }
            if(!linkList.isEmpty()){
                // 由于栈顶元素存放第一个比当前元素小的数 对应下标,则取出并给结果数组赋值
                ans[i] = linkList.peek();
            }else{
                ans[i] = -1;
            }
            linkList.push(i);
        }

        /*for (int i = 0; i < ans.length; i++) {
            System.out.print(ans[i]+" ");
        }*/
        return ans;
    }

3:LeetCode 42. 接雨水

 /**
     * LeetCode 42. 接雨水
     * 给定n个非负整数表示每个宽度为1的柱子的高度图,计算按此排序的柱子,下雨后能接多少雨水。
     * @param height
     * @return
     */
    public static int trapWater(int[] height) {
        Deque<Integer> linkList = new LinkedList<>();
        // 总雨水量
        int ans = 0;
        int n = height.length;
        for (int i = 0; i <n ; i++) {
            // 当前柱子作为右柱子,栈顶元素作为中间柱,中间柱子前面作为左柱,只能接左右两柱最低柱子高度的水
            while(!linkList.isEmpty() && height[linkList.peek()] <= height[i]){
                // 右柱比栈顶更高,才能接水。否则的话,就是满足单调递减栈的,那么我们继续入栈。
                int top = linkList.pop();// 拿出前一个柱子
                if(linkList.isEmpty()){
                    // 如果这根柱子后,前面没有元素,那就接不了雨水了,因为接雨水的话,至少需要左右两边都有柱子才行。
                    break;
                }
                // 记录一下拿到的这根柱子的左边那根柱子的高度
                int left = linkList.peek();
                // 根据柱状图推算宽度
                int currWidth = i-left-1;
                int currHeight = Math.min(height[left],height[i]) - height[top];
                ans += currHeight*currWidth;
            }
            linkList.push(i);//经过上面一顿操作之后,咱们的栈又满足单调性了,于是将当前元素的下标入栈。
        }
        return ans;
    }

参考资料
单调栈图文详解

你可能感兴趣的:(力扣算法冲刺,算法,leetcode)