原题链接

    image.png

    其实对于每根柱子,都有一个独属于它的矩形。
    矩形的宽也就是柱子本身的高度,长度则由从这根柱子开始往右第一根比它矮的柱子的位置r[i]和从这根柱子开始往左第一根比它矮的柱子的位置l[i]决定。
    所以题目变成了:对于每根柱子i,确定从这根柱子开始往右第一根比它矮的柱子的位置r[i]和从这根柱子开始往左第一根比它矮的柱子的位置l[i],典型的单调栈的题目。
    数组r每个元素要初始为n,数组l每个元素要初始为-1,这是因为对于某根柱子,它左边或者右边没有比他矮的柱子,那么此时属于这根柱子独有的矩形的长可以一直延申至左边界或者右边界。
    最后,对于每根柱子确定的矩形,找到面积最大的那个即可。

    AC代码如下:

    1. class Solution {
    2. public:
    3. int largestRectangleArea(vector<int>& heights) {
    4. stack<int> s;
    5. int n = heights.size(), ans = -1;
    6. int r[n], l[n];
    7. fill(l, l + n, -1);
    8. fill(r, r + n, n);
    9. for (int i = 0; i < n; i++)
    10. {
    11. while (s.size() > 0 && heights[i] < heights[s.top()])
    12. r[s.top()] = i, s.pop();
    13. s.push(i);
    14. }
    15. s = stack<int>();
    16. for (int i = n - 1; i >= 0; i--)
    17. {
    18. while (s.size() > 0 && heights[i] < heights[s.top()])
    19. l[s.top()] = i, s.pop();
    20. s.push(i);
    21. }
    22. for (int i = 0; i < n; i++)
    23. ans = max(ans, (r[i] - l[i] - 1) * heights[i]);
    24. return ans;
    25. }
    26. };