代码随想录算法训练营第四十九天| 42. 接雨水 、84.柱状图中最大的矩形

avatar
作者
猴君
阅读量:0

42. Trapping Rain Water

双指针暴力解法

每一列雨水的高度,取决于,该列 左侧最高的柱子和右侧最高的柱子中最矮的那个柱子的高度,即: min(lHeight, rHeight) - height

class Solution:     def trap(self, height):         if len(height) <= 2:             return 0          max_left = [0] * len(height)         max_right = [0] * len(height)         n = len(height)          # 记录每个柱子左边柱子最大高度         max_left[0] = height[0]         for i in range(1, n):             max_left[i] = max(height[i], max_left[i - 1])          # 记录每个柱子右边柱子最大高度         max_right[n - 1] = height[n - 1]         for i in range(n - 2, -1, -1):             max_right[i] = max(height[i], max_right[i + 1])          # 求和         sum_ = 0         for i in range(n):             count = min(max_left[i], max_right[i]) - height[i]             if count > 0:                 sum_ += count          return sum_         

单调栈解法:

一旦发现添加的柱子高度大于栈头元素了,此时就出现凹槽了,栈头元素就是凹槽底部的柱子,栈头第二个元素就是凹槽左边的柱子,而添加的元素就是凹槽右边的柱子。

遇到相同的元素,更新栈内下标,就是将栈里元素(旧下标)弹出,将新元素(新下标)加入栈中,因为我们要求宽度的时候 如果遇到相同高度的柱子,需要使用最右边的柱子来计算宽度。

class Solution:     def trap(self, height):         result = 0         stack = [0]         for i in range(1, len(height)):             while len(stack) > 0 and height[i] > height[stack[-1]]:                 mid_height = stack.pop()                 if stack:                     h = min(height[i], height[stack[-1]]) - height[mid_height]                     w = i - stack[-1] - 1                     v = h*w                     result += v             stack.append(i)         return result

84. Largest Rectangle in Histogram

只有栈里从大到小的顺序,才能保证栈顶元素找到左右两边第一个小于栈顶元素的柱子。

所以本题单调栈的顺序正好与接雨水反过来。

此时大家应该可以发现其实就是栈顶和栈顶的下一个元素以及要入栈的三个元素组成了我们要求最大面积的高度和宽度

# 单调栈精简 class Solution:     def largestRectangleArea(self, heights: List[int]) -> int:         heights.insert(0, 0)         heights.append(0)         stack = [0]         result = 0         for i in range(1, len(heights)):             while stack and heights[i] < heights[stack[-1]]:                 mid_height = heights[stack[-1]]                 stack.pop()                 if stack:                     # area = width * height                     area = (i - stack[-1] - 1) * mid_height                     result = max(area, result)             stack.append(i)         return result 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!