代码随想录算法训练营第六十天| 单调栈 84 柱状图中最大的矩形
LeetCode 84 柱状图中最大的矩形
题目: 84.柱状图中最大的矩形
本题和接雨水问题在原理上有不少相似之处
暴力法:
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int sum = 0;
for(int i = 0; i <heights.size(); i++) {
int left = i;
int right = i;
for(; left >= 0; left--) {
if(heights[left] < heights[i]) break;
}
for(; right < heights.size(); right++) {
if(heights[right] < heights[i]) break;
}
int w = right - left - 1;
int h = height[i];
sum = max(sum, w * h);
}
return sum;
}
};
双指针法:
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;
}
};
单调栈:
接雨水是找每个柱子左右两边第一个大于该柱子高度的柱子,本题是找每个柱子左右两边第一个小于该柱子高度的柱子,栈顶和栈顶的下一个元素以及要入栈的三个元素组成要求最大面积的高度和宽度
完整代码:
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int result = 0;
stack<int> st;
heights.insert(heights.begin(), 0); // 数组头部加入元素0
heights.push_back(0); // 数组尾部加入元素0
st.push(0);
// 第一个元素已经入栈,从下标1开始
for (int i = 1; i < heights.size(); i++) {
if (heights[i] > heights[st.top()]) { // 情况一
st.push(i);
} else if (heights[i] == heights[st.top()]) { // 情况二
st.pop();
st.push(i);
} else { // 情况三
while (!st.empty() && heights[i] < heights[st.top()]) {
int mid = st.top();
st.pop();
if (!st.empty()) {
int left = st.top();
int right = i;
int w = right - left - 1;
int h = heights[mid];
result = max(result, w * h);
}
}
st.push(i);
}
}
return result;
}
};