题目描述
Problem: 84. 柱状图中最大的矩形
解法一:暴力解法
解题方法
采用暴力遍历方式,分别以每个柱子为高度,找到以柱子i
为高度的合法矩阵区域。此时的矩阵区域的条件一定是左边界高度大于等于柱子i,右边界高度大于等于柱子i,每遇到柱子i大于左右边界的时候,就退出探查,计算结果。
C++
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int res = 0, n = heights.size();
for(int i = 0; i < n; i++) {
int left, right;
for(left = i; left >= 0; left--) { // 向左探寻,找到左边界
if(heights[left] < heights[i])
break;
}
for(right = i; right < n; right++) { // 向右探寻,向右探寻找到右边界
if(heights[i] > heights[right])
break;
}
int w = right - left - 1; // 取边界中间的面积
int h = heights[i];
res = max(res, w * h);
}
return res;
}
};
此时时间复杂度为 O ( n 2 ) O(n^2) O(n2)
Python
class Solution:
def largestRectangleArea(self, heights: List[int]) -> int:
n = len(heights)
res = 0
for i in range(n):
left = right = i
while left >= 0 and heights[left] >= heights[i]:
left -= 1
while right < n and heights[right] >= heights[i]:
right += 1
w = right - left - 1
h = heights[i]
res = max(res, h * w)
return res
会超时
解法二:单调栈
整体思路与 42. 接雨水 相近,区别在于接雨水的时候,单调栈是按从栈底到栈顶单调不增顺序存储,而本题要找到最大勾画面积,可以让每次到达一个比栈顶元素更低的长方体时,触发求面积的计算,也就是每当遇到右边第一个比自己更小的元素时,计算一次。因此,本体可按照单调栈从栈底到栈顶,单调不减的次序存储。
解题方法
采用单调栈的方式,自栈底到栈顶,按照单调不减的次序存储元素。每当第一次遇到比栈顶元素更小的数,则将栈顶元素弹栈,计算高度差和间隔差,得到面积,再和已有结果比较取最大值。
注意:由于有可能输入数组会是单调递增数组,那么如果按照原数组进行操作,就不能找到结果。因此,需要再数组末尾添加一个0,来保证可以触发计算。同时,为了防止第一个入栈元素就会被弹出,在后续还要进行求间隔差操作,为了避免空栈导致不能得到st.top()
,因此再数组头需要再插入一个0元素,让第一个数就进行运算时可正常运算不需要单独再写代码,统一了操作。
复杂度
- 时间复杂度:
O ( n ) O(n) O(n)
- 空间复杂度:
O ( n ) O(n) O(n)
Code
C++
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
heights.insert(heights.begin(), 0); // 首端加0,为了避免计算到只剩一个元素时,出现空栈现象,加一个0
heights.push_back(0); // 尾端加0,保证升序序列也可以被正常计算
int n = heights.size(), res = 0;
stack<int> st;
st.push(0);
for(int i = 1; i < n; i++) {
while(heights[st.top()] > heights[i]) {
int mid = st.top(); st.pop(); // 获取左侧的位置下标并从栈中弹出
int w = i - st.top() - 1; // 获取中间的这段长度
int h = heights[mid]; // 获取高度
res = max(res, h * w);
}
st.push(i);
}
return res;
}
};
Python
class Solution:
def largestRectangleArea(self, heights: List[int]) -> int:
heights.insert(0, 0)
heights.append(0)
stack = [0]
res, n = 0, len(heights)
for i in range(1, n):
while stack and heights[stack[-1]] > heights[i]:
mid = stack[-1]
stack.pop()
if stack:
w = i - stack[-1] - 1
h = heights[mid]
res = max(res, h * w)
stack.append(i)
return res
参考文章:84. 柱状图中最大的矩形