題目

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 height = [2,1,5,6,2,3],
return 10.

思路

我們通過一個棧記錄上升的柱子，如果如果下降的柱子，可以開始計算棧頂和之前柱子構建的矩形的面積。棧儲存的是柱子的下標，而不是柱子的高度，目的是方便計算矩形的面積。遇到上升的柱子，就把柱子對應的下標壓入棧。

程式碼

/*---------------------------------------
*   日期：2015-05-13
*   作者：SJF0115
*   題目: 84.Largest Rectangle in Histogram
*   網址：https://leetcode.com/problems/largest-rectangle-in-histogram/
*   結果：AC
*   來源：LeetCode
*   部落格：
-----------------------------------------*/
 

#include <iostream>
#include <stack>
#include <vector>
using namespace std;

class Solution {
public:
    int largestRectangleArea(vector<int>& height) {
        int maxArea = 0;
        int size = height.size();
        if(size <= 0){
            return maxArea;
        }//if
 

        // 下標
        stack<int> indexStack;
        int top,width;
        for(int i = 0;i < size;++i){
            // 棧空或上升序列 壓入棧
            if(indexStack.empty() || height[indexStack.top()] <= height[i]){
                indexStack.push(i);
            }//if
            // 一旦下降了計算面積
            else{
                top = indexStack.top();
                indexStack.pop();
                // 棧為空 表示從第一個到當前的最低高度
                width = indexStack.empty() ? i : (i - indexStack.top() - 1);
                maxArea = max(maxArea,height[top] * width);
                // 保持i的位置不變
                --i;
            }//else
        }//for
        // 計算剩餘上升序列面積
        while(!indexStack.empty()){
            top = indexStack.top();
            indexStack.pop();
            width = indexStack.empty() ? size : (size - indexStack.top() - 1);
            maxArea = max(maxArea,height[top] * width);
        }//while
        return maxArea;
    }
};

int main(){
    Solution s;
    //vector<int> height = {2,1,5,6,2,3};
    //vector<int> height = {2,1};
    //vector<int> height = {1,2,3};
    //vector<int> height = {2,1,2};
    vector<int> height = {4,2,0,3,2,5};
    cout<<s.largestRectangleArea(height)<<endl;
    return 0;
}

執行時間