题目描述
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
思路
1.暴力法:
我们需要找到最大的面积,我们可以枚举宽,使用两重循环控制矩形的宽度,然后在宽度内找最小高度,再相乘就是矩形面积,然后不断更新矩形的面积,直到遍历完所有左边界。暴力法的时间复杂度高,一般不提倡使用。
class Solution {
public int largestRectangleArea(int[] heights) {
int n = heights.length;
int ans = 0;
// 枚举左边界
for (int left = 0; left < n; ++left) {
int minHeight = Integer.MAX_VALUE;
// 枚举右边界
for (int right = left; right < n; ++right) {
// 确定高度
minHeight = Math.min(minHeight, heights[right]);
// 计算面积
ans = Math.max(ans, (right - left + 1) * minHeight);
}
}
return ans;
}
}
2.利用单调栈
如上图,遍历到位置7遍历完,栈内还有两个元素,我们此时默认在柱子位置8处加一个高度为0的柱子,然后进行比较,将7弹出栈,且area=(8-3-1)*3=12;再弹出3,且area=(8-(-1)-1)*2=16,最后栈空,开始遍历最大值。
class Solution {
public int largestRectangleArea(int[] heights) {
int n = heights.length;
int[] left = new int[n];//存放左边界
int[] right = new int[n];//存放右边界
Stack<Integer> mono_stack = new Stack<Integer>();
for (int i = 0; i <= n; i++) {
int h=(i==n?0:heights[i]);//我们遍历完所有元素后栈里还剩下元素,就需要默认在柱子的最后一列添加一个高度为0的柱子
while (!mono_stack.isEmpty() && heights[mono_stack.peek()] >= h) {//如果栈为非空且待插入元素比栈顶元素小
right[mono_stack.peek()] = i;//确定栈顶元素的右边界
mono_stack.pop();//栈顶元素弹出
}
if(i!=n){
left[i] = (mono_stack.isEmpty() ? -1 : mono_stack.peek());//确定该位置i的左边界
mono_stack.push(i);//待插入元素比栈顶元素大,压入栈
}
}
int ans = 0;
for (int i = 0; i < n; ++i) {
ans = Math.max(ans, (right[i] - left[i] - 1) * heights[i]);//遍历所有位置求出最大面积
}
return ans;
}