这个题是求一个直方图中的最大矩形面积。
这个题到现在也没感觉完全弄懂,但是大体的思路看了别人的分析之后有了个大概的认识。
如果已知height数组是升序的,应该怎么做?
比如1,2,5,7,8
那么就是(1*5) vs. (2*4) vs. (5*3) vs. (7*2) vs. (8*1)
也就是max(height[i]*(size-i))
所以使用栈来构造升序序列,每个升序序列按照以上方法求解。
指针i遍历直方图高度,如果heights[i]比当前栈顶元素高,压栈,如果比当前栈顶元素低,那么说明栈顶元素是一个升序序列的又边界,也就是最大的一个,按照上面那个例子来说,就是计算8*1,然后8弹出栈,如果heights[i]还是比当前栈顶元素小,计算7*2,以此类推。如果heights[i]是6,大于现在的栈顶元素5,那么又构成升序序列了,6压栈。继续循环。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int res = 0;
stack<int> s;
heights.push_back(0);
for (int i = 0; i < heights.size(); ++i) {
while (!s.empty() && heights[s.top()] >= heights[i]) {
int cur = s.top(); s.pop();
res = max(res, heights[cur] * (s.empty() ? i : (i - s.top() - 1)));
}
s.push(i);
}
return res;
}
};