代码随想录day59 单调栈

代码随想录day59 单调栈

题84 柱状图中最大的矩形

1,本题和接雨水思路差不多,接雨水是找左右两边高的,本题需要找左右两边矮的,这样就可以找到当前高度的柱子可以持续多久(持续的宽度)。
2,栈中存放元素的下标,且从栈头到栈尾应该单调递减,这样才能使栈头的下一个元素是比栈头元素小且距离最近的一个。
3,遍历元素时依然分3中情况,遍历元素> / = / <栈头元素, 分别处理。
4,为了防止非0 数组中存在元素找不到更小元素的情况,将原数组的最前和最后分别扩容一个0。

class Solution {
    public int largestRectangleArea(int[] heights) {
        Deque<Integer> stack = new LinkedList<>();
        int result = 0;
        //扩展原数组:在原数组的头和尾各加一个元素0,为了使数组中最小的元素(如果非0)找到更小的元素
        int[] newHeights = new int[heights.length + 2];
        newHeights[0] = 0;
        newHeights[heights.length + 1] = 0;
        for(int i = 1; i <= heights.length; i++) {
            newHeights[i] = heights[i - 1];
        }
        System.out.println(Arrays.toString(newHeights));

        //将新数组中第一个元素下标加入栈中,栈中从栈头到栈尾递减
        stack.push(0);
        for(int i = 1; i < newHeights.length; i++) {
            if(newHeights[i] > newHeights[stack.peek()]) { // 遍历元素大于栈顶元素,直接入栈
                stack.push(i);
            } else if(newHeights[i] == newHeights[stack.peek()]) {//相同时可以加也可以不加,结果一样
                stack.pop();
                stack.push(i);
            } else {
                while(!stack.isEmpty() && newHeights[i] < newHeights[stack.peek()]) { //找到右边第一个小的
                    int mid = stack.pop();
                    if(!stack.isEmpty()) {
                        int left = stack.peek(); //左边第一个小的一定是栈顶的下一个
                        int right = i;
                        int w = i - left - 1; //高度为height[mid]的柱子持续的宽度
                        result = Math.max(result, w * newHeights[mid]);
                    }
                }
                stack.push(i); //注意一定要将i入栈。
            }   
        }
        return result;
    }
}

你可能感兴趣的:(leetcode,数据结构,算法)