原作链接:
转自: 【柱状图中最大的矩形】单调栈入门,使用单调栈快速寻找边界.
题目描述:
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。求在该柱状图中,能够勾勒出来的矩形的最大面积。
测试样例:
示例 1:
输入: heights = [2,1,5,6,2,3]
输出: 10
解释: 最大的矩形为图中红色区域,面积为 10
示例 2:
输入: heights = [2,4]
输出: 4
提示:
- 1 <= heights.length <=105
- 0 <= heights[i] <= 104
单调栈:
-
单调栈分为单调递增栈和单调递减栈
1.1. 单调递增栈即栈内元素保持单调递增的栈
1.2. 同理单调递减栈即栈内元素保持单调递减的栈 -
操作规则(下面都以单调递增栈为例)
2.1. 如果新的元素比栈顶元素大,就入栈
2.2. 如果新的元素较小,那就一直把栈内元素弹出来,直到栈顶比新元素小 -
加入这样一个规则之后,会有什么效果
3.1. 栈内的元素是递增的
3.2. 当元素出栈时,说明这个新元素是出栈元素向后找第一个比其小的元素举个例子,配合下图,现在索引在 6 ,栈里是 1 5 6 。 接下来新元素是 2 ,那么 6 需要出栈。 当 6 出栈时,右边 2 代表是 6 右边第一个比 6 小的元素。
3.3. 当元素出栈后,说明新栈顶元素是出栈元素向前找第一个比其小的元素
当 6 出栈时,5 成为新的栈顶,那么 5 就是 6 左边第一个比 6 小的元素。
-
代码模板
stack<int> st; for(int i = 0; i < nums.size(); i++) { while(!st.empty() && st.top() > nums[i]) { st.pop(); } st.push(nums[i]); }
-
画图理解
思路
- 对于一个高度,如果能得到向左和向右的边界
- 那么就能对每个高度求一次面积
- 遍历所有高度,即可得出最大面积
- 使用单调栈,在出栈操作时得到前后边界并计算面积
题解:
单调栈
@Override
class Solution {
public:
int largestRectangleArea(vector<int>& heights)
{
int ans = 0;
vector<int> st;
heights.insert(heights.begin(), 0);
heights.push_back(0);
for (int i = 0; i < heights.size(); i++)
{
while (!st.empty() && heights[st.back()] > heights[i])
{
int cur = st.back();
st.pop_back();
int left = st.back() + 1;
int right = i - 1;
ans = max(ans, (right - left + 1) * heights[cur]);
}
st.push_back(i);
}
return ans;
}
};
/*
作者:ikaruga
链接:https://leetcode-cn.com/problems/largest-rectangle-in-histogram/solution/84-by-ikaruga/
来源:力扣(LeetCode)
*/