leetcode 84. 柱状图中最大的矩形


题目描述:

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

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

输入:heights = [2,1,5,6,2,3]
输出:10
解释:最大的矩形为图中红色区域,面积为 10

解题思路:

看题目本身,跟经典的接雨水有点类似。但不同点是接雨水的高度是由左右边界的最小值决定,这题的高度则是由每个柱子本身决定。

1. 暴力解法:自己能想到的解法。由当前柱子向左右两边延伸,只要满足高度大于当前柱子的,就能继续延伸,直到遇到高度小于当前柱子或到达边界。这实际上是以每个柱子为中心,向左右去寻找当前柱子的边界,将边界作为矩形宽度,柱子高度为矩形长度。最后取面积最大的矩形为最终结果。时间复杂度O(n^2),会超时。

2. 单调栈:大四算法课上单独讲过这类算法。实际是拿空间换时间,维护一个单调递增的栈,一定程度上是保证了左边界,再通过遍历的方式寻找右边界。始终维护一个单调递增的栈,遇到小于栈顶元素高度的柱子,则需要将栈顶元素出栈,也就意味着对于当前栈顶的元素来说,找到了其左右边界,左边界就为栈的下一个元素(若不存在则边界为-1),右边界为当前遍历到元素。当遍历结束后,需要清空栈,此时的右边界的确定的,为数组的长度,需要确定的是左边界,同样为栈的下一个元素(若不存在则边界为-1)。

代码:

解法一:

class Solution {
public:
    int largestRectangleArea(vector& heights) {
        int n = heights.size();
        int ans = 0;
        for(int i=0; i=0 && heights[i] <= heights[l]) {
                l--;
            }
            while(r=n)
                r = n-1;
            else if(r != i)
                r = r-1;
            
            if(l<0)
                l=0;
            else if(l != i)
                l = l+1;
            int area = heights[i]*(r-l+1);
            ans = max(ans, area);
        }
        return ans;
    }
};

解法二:

class Solution {
public:
    int largestRectangleArea(vector& heights) {
        int n = heights.size();
        stack s;
        int ans = 0;
        for(int i=0; i heights[i]) {
                int tmp = s.top();
                s.pop();
                int length = heights[tmp];
                int weight = i;
                if(!s.empty())
                    weight = i-s.top()-1;
                ans = max(ans, length*weight);
            }
            s.push(i);
        }
        
        int length = 0;
        int weight = n;
        while(!s.empty()) {
            int tmp = s.top();
            s.pop();
            length = heights[tmp];
            weight = n;
            if(!s.empty())
                weight = n-s.top()-1;
            ans = max(ans, length*weight);
        }
        
        return ans;
        
    }
};