[栈] leetcode 84 Largest Rectangle in Histogram

problem:https://leetcode.com/problems/largest-rectangle-in-histogram

        经典单调栈题目。维护递增的单调栈,当发现当前数字比栈顶要小的时候,此时栈顶元素是最大的(大于栈里的下一个元素,也大于当前元素),所以可以计算以当前栈顶元素为高的矩形面积,并比较是不是最大的。

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        heights.push_back(0);
        int n = heights.size();
        stack<int> sta;
        int res = 0;
        
        for(int i = 0;i < n;i++)
        {
            while(!sta.empty() && heights[i] <= heights[sta.top()])
            {
                int h = heights[sta.top()];
                sta.pop();
                int left = sta.empty() ? -1 : sta.top();
                int s = h * (i - left - 1);
                res = max(res, s);
                
            }
     //       cout << res << " ";
            sta.push(i);
        }
        return res;
    }
};

猜你喜欢

转载自www.cnblogs.com/fish1996/p/11335097.html