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<int>& heights) {
        int n = heights.size();
        int ans = 0;
        for(int i=0; i<n; i++) {
            int l = i;
            int r = i;
            while(l>=0 && heights[i] <= heights[l]) {
                l--;
            }
            while(r<n && heights[i] <= heights[r]) {
                r++;
            }
            if(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<int>& heights) {
        int n = heights.size();
        stack<int> s;
        int ans = 0;
        for(int i=0; i<n; i++) {
            while(!s.empty() && heights[s.top()] > 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;
        
    }
};

  

posted @ 2022-03-15 23:29  Fzu_LJ  阅读(34)  评论(0编辑  收藏  举报