代码随想录算法训练营第六十天|84.柱状图中最大的矩形

最新推荐文章于 2024-07-13 23:05:26 发布

一楼二栋

最新推荐文章于 2024-07-13 23:05:26 发布

阅读量346

点赞数 6

分类专栏：代码随想录文章标签：算法 leetcode c++

本文链接：https://blog.csdn.net/m0_71413464/article/details/136320708

版权

代码随想录专栏收录该内容

47 篇文章 0 订阅

订阅专栏

文章介绍了如何使用C++的动态规划方法，借助栈来解决LeetCode上的84号问题——在柱状图中找到面积最大的矩形。通过遍历和比较高度，维护栈来找到每个矩形的左边界和高度，计算并更新最大面积。

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

题目：84.柱状图中最大的矩形

文章链接：代码随想录

视频链接：LeetCode:84.柱状图中最大的矩形

题目链接：力扣题目链接

图释：

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        // 输出结果
        int result = 0; 
        stack<int> st;
        heights.insert(heights.begin(), 0); // 数组头部加入元素0
        heights.push_back(0); // 数组结尾加入元素0  为了解决本身数组为递增时，最后一个元素进行截断
        st.push(0);

        // 第一个元素已经入栈，从下标1开始
        for(int i=1; i<heights.size(); i++){
            if(heights[i]>heights[st.top()]){ // 当前元素大于栈口元素，则添加到栈中
                st.push(i);
            }else if(heights[i]==heights[st.top()]){ // 当前元素等于栈口元素，则添加到栈中也可以将栈口元素弹出再加进去
                // st.pop();
                // st.push(i);
                st.push(i);
            }else{
                while(!st.empty() && heights[i]<heights[st.top()]){  // 循环拿出栈中的元素跟当前的元素进行比较，直到当前元素小于等于栈口元素
                    int mid = st.top(); // 中间为栈顶元素，作为高
                    st.pop(); 
                    if(!st.empty()){  // 判断栈内是否还有元素
                        int left = st.top(); //左边的高度
                        int right = i; 
                        int w = right - left -1; 右减左再减1作为宽
                        int h = heights[mid];
                        result = max(result, w*h);
                    }
                }
                st.push(i);
            }
        }
        return result;
    }
};