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

文章目录

42.接雨水

双指针

其实感觉和双指针关系不大(如果是优化前的版本确实是双指针),因为每一个木块位置能存放的水的数量取决于他所处位置的水平线,水平线则是由它的向左找到的最大值和向右找到的最大值中的短板,拿水平线减去木板高度就是这个位置能找到的最大储水。

copy两次height生成每个位置的向左最大高度和向右最大高度(用height作为初值是因为这个高度至少要大于当前木板高度)。然后对于每个木板,他的向左最大高度可以通过前一个位置的向左最大高度和自身高度比较取大的值,向右同理。

python 复制代码
class Solution:
    def trap(self, height: List[int]) -> int:
        l = len(height)
        max_left = height[:]
        max_right = height[:]
        ans = 0
        for i in range(1, l):
            max_left[i] = max(max_left[i-1], height[i])
        for i in range(l-2, -1, -1):
            max_right[i] = max(max_right[i+1], height[i])
        for i in range(1, l-1):
            ans += min(max_left[i], max_right[i]) - height[i]
        return ans

单调栈

用单调栈来保存遍历过的木板,保持栈底到栈头从大到小的顺序,如果新的木板大于栈头的木板,则出栈,该木板决定了现在这个木桶的桶底的高度,然后还需要再判断一次栈是否为空,只有不为空的时候才能再取到左木板,使用这三块木板算出当前桶的存水量。

因为栈顶的木板是确定桶底高度,所以右木板必须高于它的时候才可以进行使用。

python 复制代码
class Solution:
    def trap(self, height: List[int]) -> int:
        st = [0]
        ans = 0
        for i in range(1, len(height)):
            while st and height[i] > height[st[-1]]:
                bottom = st.pop()
                if st:
                    ans += (min(height[st[-1]], height[i]) - height[bottom]) * (i - st[-1] - 1)
            st.append(i)
        return ans

84.柱状图中最大的矩形

和上一题是反过来的,这一题是对每个矩形往两边去找第一个小于他的值,然后在这两个值的范围内算出这个高度的可以围成的最大面积,所以单调栈栈底到栈头是从小到大,方便找到左边第一个小于它的索引。由于不像上一题最左右的两块不能参与到计算中,这一题最左右的矩形是需要计算它的可能围成面积的,所以需要额外在头尾加上一个0。

python 复制代码
class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        st = [0]
        heights.insert(0, 0)
        heights.append(0)
        ans = 0
        for i in range(1, len(heights)):
            while st and heights[i] < heights[st[-1]]:
                height = st.pop()
                if st:
                    ans = max(ans, heights[height]*(i-st[-1]-1))
            st.append(i)
        return ans
相关推荐
良木生香24 分钟前
【鼠鼠优选算法-双指针】003:快乐数 & 004:盛水最多的容器
算法
Cx330❀25 分钟前
【优选算法必刷100题】第41-42题(模拟):Z 字形变换,外观数列
c++·算法
沃尔特。27 分钟前
直流无刷电机FOC控制算法
c语言·stm32·嵌入式硬件·算法
CW32生态社区28 分钟前
CW32L012的PID温度控制——算法基础
单片机·嵌入式硬件·算法·pid·cw32
Cx330❀29 分钟前
【优选算法必刷100题】第038题(位运算):消失的两个数字
开发语言·c++·算法·leetcode·面试
漫随流水30 分钟前
leetcode回溯算法(93.复原IP地址)
数据结构·算法·leetcode·回溯算法
燃于AC之乐31 分钟前
我的算法修炼之路--5——专破“思维陷阱”,那些让你拍案叫绝的非常规秒解
c++·算法·贪心算法·bfs·二分答案·扩展域并查集·动态规划(最长上升子序列)
艾莉丝努力练剑31 分钟前
【优选算法必刷100题】第021~22题(二分查找算法):山脉数组的峰顶索引、寻找峰值
数据结构·c++·算法·leetcode·stl
艾莉丝努力练剑33 分钟前
【优选算法必刷100题】第007~008题(双指针算法):三数之和、四数之和问题求解
linux·算法·双指针·优选算法
gihigo19982 小时前
希尔伯特-黄变换(HHT)完整MATLAB实现
人工智能·算法·matlab