代码随想录day59 单调栈

最新推荐文章于 2024-11-13 16:35:03 发布

shiliuhua05

最新推荐文章于 2024-11-13 16:35:03 发布

阅读量284

点赞数

分类专栏： leetcode 文章标签：数据结构算法

本文链接：https://blog.csdn.net/shiliuhua05/article/details/128933289

版权

leetcode 专栏收录该内容

50 篇文章 1 订阅

订阅专栏

博客探讨了如何使用单调栈解决柱状图中寻找最大矩形的问题，类似于接雨水，通过栈来找到每个柱子能构成矩形的宽度，确保栈内元素下标单调递减，从而确定当前高度的持续宽度。

摘要由CSDN通过智能技术生成

代码随想录day59 单调栈

题84 柱状图中最大的矩形

1，本题和接雨水思路差不多，接雨水是找左右两边高的，本题需要找左右两边矮的，这样就可以找到当前高度的柱子可以持续多久(持续的宽度)。
2，栈中存放元素的下标，且从栈头到栈尾应该单调递减，这样才能使栈头的下一个元素是比栈头元素小且距离最近的一个。
3，遍历元素时依然分3中情况，遍历元素> / = / <栈头元素，分别处理。
4，为了防止非0 数组中存在元素找不到更小元素的情况，将原数组的最前和最后分别扩容一个0。

class Solution {
    public int largestRectangleArea(int[] heights) {
        Deque<Integer> stack = new LinkedList<>();
        int result = 0;
        //扩展原数组：在原数组的头和尾各加一个元素0，为了使数组中最小的元素（如果非0）找到更小的元素
        int[] newHeights = new int[heights.length + 2];
        newHeights[0] = 0;
        newHeights[heights.length + 1] = 0;
        for(int i = 1; i <= heights.length; i++) {
            newHeights[i] = heights[i - 1];
        }
        System.out.println(Arrays.toString(newHeights));

        //将新数组中第一个元素下标加入栈中,栈中从栈头到栈尾递减
        stack.push(0);
        for(int i = 1; i < newHeights.length; i++) {
            if(newHeights[i] > newHeights[stack.peek()]) { // 遍历元素大于栈顶元素，直接入栈
                stack.push(i);
            } else if(newHeights[i] == newHeights[stack.peek()]) {//相同时可以加也可以不加，结果一样
                stack.pop();
                stack.push(i);
            } else {
                while(!stack.isEmpty() && newHeights[i] < newHeights[stack.peek()]) { //找到右边第一个小的
                    int mid = stack.pop();
                    if(!stack.isEmpty()) {
                        int left = stack.peek(); //左边第一个小的一定是栈顶的下一个
                        int right = i;
                        int w = i - left - 1; //高度为height[mid]的柱子持续的宽度
                        result = Math.max(result, w * newHeights[mid]);
                    }
                }
                stack.push(i); //注意一定要将i入栈。
            }   
        }
        return result;
    }
}