力扣 84. 柱状图中最大的矩形

题目来源:https://leetcode.cn/problems/largest-rectangle-in-histogram/description/

C++题解1:双指针法(来源代码随想录

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        vector<int> minLeftIndex(heights.size());
        vector<int> minRightIndex(heights.size());
        int size = heights.size();

        // 记录每个柱子 左边第一个小于该柱子的下标
        minLeftIndex[0] = -1; // 注意这里初始化,防止下面while死循环
        for (int i = 1; i < size; i++) {
            int t = i - 1;
            // 这里不是用if,而是不断向左寻找的过程
            while (t >= 0 && heights[t] >= heights[i]) t = minLeftIndex[t];
            minLeftIndex[i] = t;
        }
        // 记录每个柱子 右边第一个小于该柱子的下标
        minRightIndex[size - 1] = size; // 注意这里初始化,防止下面while死循环
        for (int i = size - 2; i >= 0; i--) {
            int t = i + 1;
            // 这里不是用if,而是不断向右寻找的过程
            while (t < size && heights[t] >= heights[i]) t = minRightIndex[t];
            minRightIndex[i] = t;
        }
        // 求和
        int result = 0;
        for (int i = 0; i < size; i++) {
            int sum = heights[i] * (minRightIndex[i] - minLeftIndex[i] - 1);
            result = max(sum, result);
        }
        return result;
    }
};

C++题解2:单调栈

本题是要找每个柱子左右两边第一个小于该柱子的柱子,所以从栈头(元素从栈头弹出)到栈底的顺序应该是从大到小的顺序!

只有栈里从大到小的顺序,才能保证栈顶元素找到左右两边第一个小于栈顶元素的柱子。

1. 自己写的较笨的方法的,是左右两边都用栈寻找第一个小于栈顶元素的柱子

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        int len = heights.size();
        stack<int> st;
        int result = 0;

        // 正向
        vector<int> res(heights.begin(), heights.end());
        st.push(0);
        for(int i = 1; i < len; i++) {
            while(!st.empty() && heights[i] < heights[st.top()]) {
                res[st.top()] = (i - st.top()) * heights[st.top()];
                st.pop();
            }
            st.push(i);
        }
        while(!st.empty()) {
            res[st.top()] = (len - st.top()) * heights[st.top()];
            st.pop();
        }

        // 反向
        vector<int> h1(heights.begin(), heights.end());
        reverse(h1.begin(), h1.end());
        vector<int> r1(h1.begin(), h1.end());
        st.push(0);
        for(int i = 1; i < len; i++) {
            while(!st.empty() && h1[i] < h1[st.top()]) {
                r1[st.top()] = (i - st.top()) * h1[st.top()];
                st.pop();
            }
            st.push(i);
        }
        while(!st.empty()) {
            r1[st.top()] = (len - st.top()) * h1[st.top()];
            st.pop();
        }
        reverse(r1.begin(), r1.end());

        // 以i列为中心,形成的最大面积
        for(int i = 0; i < len; i++) {
            result = max(result, res[i] + r1[i] - heights[i]);
        }

        return result;
    }
};

2. 栈顶和栈顶的下一个元素以及要入栈的三个元素组成了我们要求最大面积的高度和宽度(来源代码随想录

开头为什么要加元素0?

如果数组本身是降序的,例如 [8,6,4,2],在 8 入栈后,6 开始与8 进行比较,此时我们得到 mid(8),rigt(6),但是得不到 left。

因为 将 8 弹出之后,栈里没有元素了,那么为了避免空栈取值,直接跳过了计算结果的逻辑。

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        stack<int> st;
        heights.insert(heights.begin(), 0); // 数组头部加入元素0
        heights.push_back(0); // 数组尾部加入元素0
        st.push(0);
        int result = 0;
        for (int i = 1; i < heights.size(); i++) {
            while (heights[i] < heights[st.top()]) {
                int mid = st.top();
                st.pop();
                int w = i - st.top() - 1;
                int h = heights[mid];
                result = max(result, w * h);
            }
            st.push(i);
        }
        return result;
    }
};

  • 10
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值