LeetCode84. 柱状图中最大的矩形(单调栈)

LeetCode84. 柱状图中最大的矩形

给定 n 个非负整数，用来表示柱状图中各个柱子的高度。每个柱子彼此相邻，且宽度为 1 。

求在该柱状图中，能够勾勒出来的矩形的最大面积。

以上是柱状图的示例，其中每个柱子的宽度为 1，给定的高度为 [2,1,5,6,2,3]。

图中阴影部分为所能勾勒出的最大矩形面积，其面积为 10 个单位。

示例:

输入: [2,1,5,6,2,3]
输出: 10

一、暴力解法

class Solution {
    public int largestRectangleArea(int[] heights) {
        int len=heights.length;
        if(len==0) return 0;
        int res=0;
        for(int i=0;i<len;i++){
            int left=i;
            int cheight=heights[i];
            //找到左边大于等于当前整数的坐标
            while(left > 0 && heights[left-1] >= cheight){
                left--;
            }
            int right=i;
            //找到右边
            while(right < len-1 && heights[right+1] >= cheight){
                right++;
            }
            int width=right-left+1;
            res=Math.max(res,width*cheight);
        }
        return res;
    }
}

该方法直接超出时间限制，因此并不行。

二、单调栈

class Solution {
    public int largestRectangleArea(int[] heights) {
        //使用单调栈
        int len=heights.length;
        if(len==0) return 0;
        int[] newheights=new int[len+2];
        newheights[0]=0;
        newheights[newheights.length-1]=0;
        for(int i=1;i<=len;i++){
            newheights[i]=heights[i-1];
        }
        //得到了一个左右边界为0的新数组
        Stack<Integer> stack=new Stack<>();
        int res=0;
        for(int i=0;i<newheights.length;i++){
            while(!stack.isEmpty() && newheights[i] < newheights[stack.peek()]){
                //将栈顶元素弹出
                int cur=stack.pop();
                //得到栈顶元素对应的高度
                int curheight=newheights[cur];
                //弹出的栈顶元素的左侧边界就是新的栈顶元素
                int leftheight=stack.peek();
                //右侧边界就是当前元素
                int righheight=i;
                //所以矩形的宽度为
                int width=righheight-leftheight-1;
                res=Math.max(res,curheight*width);
            }
            stack.push(i);
        }
        return res;
    }
}