1. 程式人生 > >LeetCode84.柱狀圖中最大的矩形

LeetCode84.柱狀圖中最大的矩形

題解

單調棧板子題,建立一個單調遞增棧(棧底到棧頂是遞增的),棧記憶體放的陣列下標,遍歷陣列,將下標存進棧內,以樣例來說

首先棧空,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; } }