剑指 Offer II 039 | 直方图最大矩形面积
给定非负整数数组 heights ,数组中的数字用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
示例 1:
输入:heights = [2,1,5,6,2,3]
输出:10
解释:最大的矩形为图中红色区域,面积为 10
示例 2:
输入: heights = [2,4]
输出: 4
提示:
- 1 <= heights.length <=105
- 0 <= heights[i] <= 104
思路:用一个单调递增的栈保存柱子的下标。当前柱子的高度大于栈顶柱子的高度时,将当前柱子的下标入栈;当前柱子的高度小于栈顶柱子的高度时,求以栈顶柱子的高度为顶时最大矩形的面积,栈顶柱子下标出栈,直到当前柱子下标可以存入栈中。
栈顶柱子高度为顶时矩形的面积:高度就是柱子的高度;宽度是从栈顶柱子往前数,直到遇到高度更低的柱子,因为是单调栈,所以栈顶的前一个元素的高度是低于栈顶的高度的。
int largestRectangleArea(vector<int>& heights) {
stack<int> st;
st.push(-1);
int res = 0;
heights.push_back(-1);
for (int i = 0; i < heights.size(); i++) {
while (st.top() != -1 && heights[i] < heights[st.top()]) {
int height = heights[st.top()];
st.pop();
int width = i - st.top() - 1; // 获取栈顶的上一个柱子的下标 - 1
res = max(res, width * height);
}
st.push(i);
}
return res;
}