📑 题目:84. 柱状图中最大的矩形

🚀 本题 LeetCode 传送门

题目大意

给出每个直方图的高度,要求在这些直方图之中找到面积最大的矩形,输出矩形的面积。

解题思路

用单调栈依次保存直方图的高度下标,一旦出现高度比栈顶元素小的情况就取出栈顶元素,单独计算一下这个栈顶元素的矩形的高度。然后停在这里(外层循环中的 i–,再 ++,就相当于停在这里了),继续取出当前最大栈顶的前一个元素,即连续弹出 2 个最大的,以稍小的一个作为矩形的边,宽就是 2 计算面积…………如果停在这里的下标代表的高度一直比栈里面的元素小,就一直弹出,取出最后一个比当前下标大的高度作为矩形的边。宽就是最后一个比当前下标大的高度和当前下标 i 的差值。计算出面积以后不断的更新 maxArea 即可。

代码

  1. package leetcode
  2. func largestRectangleArea(heights []int) int {
  3. maxArea := 0
  4. n := len(heights) + 2
  5. // Add a sentry at the beginning and the end
  6. getHeight := func(i int) int {
  7. if i == 0 || n-1 == i {
  8. return 0
  9. }
  10. return heights[i-1]
  11. }
  12. st := make([]int, 0, n/2)
  13. for i := 0; i < n; i++ {
  14. for len(st) > 0 && getHeight(st[len(st)-1]) > getHeight(i) {
  15. // pop stack
  16. idx := st[len(st)-1]
  17. st = st[:len(st)-1]
  18. maxArea = max(maxArea, getHeight(idx)*(i-st[len(st)-1]-1))
  19. }
  20. // push stack
  21. st = append(st, i)
  22. }
  23. return maxArea
  24. }
  25. func max(a int, b int) int {
  26. if a > b {
  27. return a
  28. }
  29. return b
  30. }