代码随想录算法训练营第四十九天|接雨水、柱状图最大的矩形

接雨水 leetcode 42

思路：找到当前柱子右边比它大的第一个柱子以及左边比它大的第一个柱子，使用单调栈存放结果即可实现。

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> sta;
        sta.push(0);
        int s=0;
        for(int i=1;i<height.size();i++){
            while(!sta.empty()&&height[i]>height[sta.top()]){
                int middle=sta.top();
                sta.pop();
                if(!sta.empty()){
                    int h=min(height[i],height[sta.top()])-height[middle];
                    int w=i-sta.top()-1;
                    s+=w*h;
                }  
            }
            sta.push(i);
        }
        return s;
    }
};

柱状图中最大的矩形 leetcode 84

思路：找到在左边和右边第一个比当前柱子矮的柱子，如果没有说明可以扩展到边界，右边减去左边索引再减一就是宽，当前柱子高度为高得到面积，不断更新结果取最大值。

首尾加0，首部加0是为了防止出现数组单调递减的情况，因为进行比较操作时需要三个元素。尾部加0是为了防止出现数组单调递增的情况，如果单调递增那么就无法进行到比较操作的步骤。

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        stack<int> sta;
        heights.insert(heights.begin(),0);
        heights.push_back(0);
        sta.push(0);
        int res=0;
        for(int i=1;i<heights.size();i++){
            while(!sta.empty()&&heights[i]<heights[sta.top()]){
                int middle=sta.top();
                sta.pop();
                int h=heights[middle];
                int w=i-sta.top()-1;
                res=max(h*w,res);
            }
            sta.push(i);
        }
        return res;
    }
};