题目来源:https://leetcode.cn/problems/largest-rectangle-in-histogram/description/
C++题解1:双指针法(来源代码随想录)
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;
}
};
C++题解2:单调栈
本题是要找每个柱子左右两边第一个小于该柱子的柱子,所以从栈头(元素从栈头弹出)到栈底的顺序应该是从大到小的顺序!
只有栈里从大到小的顺序,才能保证栈顶元素找到左右两边第一个小于栈顶元素的柱子。
1. 自己写的较笨的方法的,是左右两边都用栈寻找第一个小于栈顶元素的柱子
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int len = heights.size();
stack<int> st;
int result = 0;
// 正向
vector<int> res(heights.begin(), heights.end());
st.push(0);
for(int i = 1; i < len; i++) {
while(!st.empty() && heights[i] < heights[st.top()]) {
res[st.top()] = (i - st.top()) * heights[st.top()];
st.pop();
}
st.push(i);
}
while(!st.empty()) {
res[st.top()] = (len - st.top()) * heights[st.top()];
st.pop();
}
// 反向
vector<int> h1(heights.begin(), heights.end());
reverse(h1.begin(), h1.end());
vector<int> r1(h1.begin(), h1.end());
st.push(0);
for(int i = 1; i < len; i++) {
while(!st.empty() && h1[i] < h1[st.top()]) {
r1[st.top()] = (i - st.top()) * h1[st.top()];
st.pop();
}
st.push(i);
}
while(!st.empty()) {
r1[st.top()] = (len - st.top()) * h1[st.top()];
st.pop();
}
reverse(r1.begin(), r1.end());
// 以i列为中心,形成的最大面积
for(int i = 0; i < len; i++) {
result = max(result, res[i] + r1[i] - heights[i]);
}
return result;
}
};
2. 栈顶和栈顶的下一个元素以及要入栈的三个元素组成了我们要求最大面积的高度和宽度(来源代码随想录)
开头为什么要加元素0?
如果数组本身是降序的,例如 [8,6,4,2],在 8 入栈后,6 开始与8 进行比较,此时我们得到 mid(8),rigt(6),但是得不到 left。
因为 将 8 弹出之后,栈里没有元素了,那么为了避免空栈取值,直接跳过了计算结果的逻辑。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
stack<int> st;
heights.insert(heights.begin(), 0); // 数组头部加入元素0
heights.push_back(0); // 数组尾部加入元素0
st.push(0);
int result = 0;
for (int i = 1; i < heights.size(); i++) {
while (heights[i] < heights[st.top()]) {
int mid = st.top();
st.pop();
int w = i - st.top() - 1;
int h = heights[mid];
result = max(result, w * h);
}
st.push(i);
}
return result;
}
};