法1:单调栈[原版]
O(N)+O(N)
必须掌握算法!!!
Python
class Solution:
def largestRectangleArea(self, heights: List[int]) -> int:
n = len(heights)
left_min = [-1] * n
right_min = [n] * n
stack = collections.deque()
for i in range(n):
while len(stack) > 0 and heights[stack[-1]] >= heights[i]:
stack.pop()
left_min[i] = -1 if len(stack) == 0 else stack[-1]
stack.append(i)
stack.clear()
for i in range(n-1, -1, -1):
while len(stack) > 0 and heights[stack[-1]] >= heights[i]:
stack.pop()
right_min[i] = n if len(stack) == 0 else stack[-1]
stack.append(i)
res = -1
for i in range(n):
res = max(res, (right_min[i] - left_min[i] - 1) * heights[i])
return res
Java
class Solution {
public int largestRectangleArea(int[] heights) {
int n = heights.length, res = 0;
int[] leftMin = new int[n], rightMin = new int[n];
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < n; ++i) {
while (!stack.isEmpty() && heights[stack.peek()] >= heights[i]) {
stack.pop();
}
leftMin[i] = stack.isEmpty() ? -1 : stack.peek();
stack.push(i);
}
stack.clear();
for (int i = n - 1; i >= 0; --i) {
while (!stack.isEmpty() && heights[stack.peek()] >= heights[i]) {
stack.pop();
}
rightMin[i] = stack.isEmpty() ? n : stack.peek();
stack.push(i);
}
for (int i = 0; i < n; ++i) {
res = Math.max(res, (rightMin[i] - leftMin[i] - 1) * heights[i]);
}
return res;
}
}
法2:单调栈[优化版]
O(N)+O(N)
参考答案
class Solution {
public int largestRectangleArea(int[] heights) {
int n = heights.length, res = 0;
int[] leftMin = new int[n], rightMin = new int[n];
Arrays.fill(rightMin, n); // 一定注意这次需要初始化!!!
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < n; ++i) {
while (!stack.isEmpty() && heights[stack.peek()] >= heights[i]) {
rightMin[stack.peek()] = i;
stack.pop();
}
leftMin[i] = stack.isEmpty() ? -1 : stack.peek();
stack.push(i);
}
for (int i = 0; i < n; ++i) {
res = Math.max(res, (rightMin[i] - leftMin[i] - 1) * heights[i]);
}
return res;
}
}