LeetCode84.柱狀圖中最大的矩形
阿新 • • 發佈:2019-01-03
題解
單調棧板子題,建立一個單調遞增棧(棧底到棧頂是遞增的),棧記憶體放的陣列下標,遍歷陣列,將下標存進棧內,以樣例來說
首先棧空,0直接進棧;然後因為nums[stack.peek()] > nums[1],所以0出棧了,同時記錄以num[0]為高的矩形的面積,當前遍歷到的陣列下標為i,此時i是1,k等於0出站後,棧頂元素的下標,k= stack.isEmpty() ? -1 : stack.peek(),最終,底的長度就是i-k-1。
如果陣列已經遍歷到結束了,棧內還有值,就需要將他們依次彈出,此時的i=nums.length(),k=stack.isEmpty() ? -1 : stack.peek()。
整個流程用一個maxArea變數維護,找到最大值即可
程式碼
class Solution {
public static int largestRectangleArea(int[] heights) {
if(heights == null || heights.length == 0)
return 0;
int maxArea = 0;
Stack<Integer> stack = new Stack<Integer>();
for(int i = 0;i < heights. length;i++) {
while(!stack.isEmpty() && heights[i] <= heights[stack.peek()]) {
int j = stack.pop();
int k = stack.isEmpty() ? -1 : stack.peek();
int curArea = (i - k - 1) * heights[j];
maxArea = Math.max(maxArea,curArea);
}
stack. push(i);
}
while(!stack.isEmpty()) {
int i = heights.length;
int j = stack.pop();
int k = stack.isEmpty() ? -1 : stack.peek();
int curArea = (i - k - 1) * heights[j];
maxArea = Math.max(maxArea,curArea);
}
return maxArea;
}
}