leetcode-84. Largest Rectangle in Histogram

来源:互联网 发布:java守护线程的作用 编辑:程序博客网 时间:2024/05/18 01:00

leetcode-84. Largest Rectangle in Histogram

题目:

Given n non-negative integers representing the histogram’s bar height where the width of each bar is 1, find the area of largest rectangle in the histogram.

Above is a histogram where width of each bar is 1, given height = [2,1,5,6,2,3].

The largest rectangle is shown in the shaded area, which has area = 10 unit.

For example,
Given heights = [2,1,5,6,2,3],
return 10.

这里显然的方法是用栈。问题在于栈里是数据存放什么?栈里存放的数据应该是坐标。是比当前的坐标所指定的高度小的前一个i的坐标。因为计算面积时需要按照最小的坐标来算。因此只需要考虑比当前i小的坐标就好。如果当前的坐标比之前的所有的坐标都大,那只需要将当前的坐标压入栈,如果当前坐标比栈顶坐标小则开始算计。

此外还需要注意一种情况。这样压入栈策略虽然效率比较高,但是还有可能出现面对一个递增的序列,直到指针指到数组末尾依然没有完成任何计算的情况。也就是说这样的压入栈的策略只有在出现下降的时候才计算。所以需要在最后把h赋值为0。这样就强制出现一个下降沿,从而完成之前的计算。

实际编程的时候else里巧用了i–,也可以用while替代。

答案来源

public class Solution {    public int largestRectangleArea(int[] heights) {        Stack<Integer> sta = new Stack<Integer>();        int ret = 0;        for(int i = 0 ; i <= heights.length; i++){            int h = i==heights.length ? 0 : heights[i];            if(sta.isEmpty() || h >= heights[sta.peek()]){                sta.push(i);            }else{                int t = sta.pop();                ret = Math.max(ret,heights[t] * (sta.isEmpty() ? i :i-1-sta.peek()));                i--;            }        }        return ret;    }}
0 0
原创粉丝点击