北辰集团网站建设,大学生个人简历word模板免费下载,免费域名申请教程,西部数码网站管理助手 ftp题目难度: 困难 原题链接 今天继续更新 Leetcode 的剑指 Offer#xff08;专项突击版#xff09;系列, 大家在公众号 算法精选 里回复 剑指offer2 就能看到该系列当前连载的所有文章了, 记得关注哦~ 题目描述
给定非负整数数组 heights #xff0c;数组中的数字用来表示柱状… 题目难度: 困难 原题链接 今天继续更新 Leetcode 的剑指 Offer专项突击版系列, 大家在公众号 算法精选 里回复 剑指offer2 就能看到该系列当前连载的所有文章了, 记得关注哦~ 题目描述
给定非负整数数组 heights 数组中的数字用来表示柱状图中各个柱子的高度。每个柱子彼此相邻且宽度为 1 。
求在该柱状图中能够勾勒出来的矩形的最大面积。
示例 1: 输入heights [2,1,5,6,2,3]输出10解释最大的矩形为图中红色区域面积为 10
示例 2 输入 heights [2,4]输出 4
提示
1 heights.length 10^50 heights[i] 10^4
题目思考
如何优化时间复杂度?
解决方案
思路
分析题目, 最容易想到的思路是暴力两层循环, 具体做法如下: 外层循环遍历每个柱子, 记录其高度 h内层向左右两边扩展, 直到超出数组范围或低于当前柱子, 记录对应的下标 l 和 r此时即为使用当前柱子高度时的矩形, 计算其面积 (r-l-1)*h 并更新最终结果这样遍历完成后就覆盖了所有可能的矩形, 其最大面积即为所求 暴力算法虽然简单, 但其时间复杂度达到了 O(N^2), 根据题目输入规模, 肯定会超时, 如何优化呢?我们的目的是计算所有矩形的面积, 而高度是已知的, 如何快速得到每个柱子的左右边界呢?由于柱子的左右边界需低于当前柱子, 而且我们既需要知道高度, 又需要知道宽度(下标), 所以这里可以采用单调栈存下标的方式实现, 具体做法如下: 单调栈存柱子下标, 且保证从栈顶到栈底的高度递减遍历某个柱子时, 先将其与栈顶高度比较 如果栈顶更高, 则将栈顶弹出, 保证单调性, 同时栈顶对应的矩形面积也可以计算了, 其右边界就是当前柱子, 左边界就是栈顶下面一个元素或者-1(对应栈顶左边没有更低柱子的情况), 右-左-1就是矩形的宽否则就退出循环, 将当前柱子压入栈中, 此时栈顶到栈底的高度仍是递减的 遍历完所有柱子后, 栈中仍可能存在一些柱子, 此时说明这些柱子右边没有更高的柱子, 其右边界就是数组长度, 左边界和上面情况一样, 依次将其弹出并计算面积最终结果就是上述所有矩形面积的最小值 利用单调栈, 我们使用和暴力算法一样的思路计算所有矩形面积, 但却将时间复杂度成功降低到了 O(N), 因为每个柱子只需要处理两次(一次入栈一次出栈)下面的代码就对应了上面的整个过程, 并且有详细的注释, 方便大家理解
复杂度
时间复杂度 O(N): 数组每个元素最多处理 2 遍 (压入和弹出栈)空间复杂度 O(N): 栈最多存 N 个元素
代码
class Solution:def largestRectangleArea(self, heights: List[int]) - int:# stack存储柱子的下标, 且其高度满足从栈顶到栈底递减stack []res 0for r, h in enumerate(heights):while stack and heights[stack[-1]] h:# 栈顶高度大于当前高度, 可以计算栈顶柱子对应的矩形面积了# 栈顶柱子的右边界r就是当前下标, 左边界l是上一个栈顶或-1(上一个栈顶不存在时)ch heights[stack.pop()]l -1 if not stack else stack[-1]# 宽*高res max(res, (r - l - 1) * ch)stack.append(r)# 如果遍历结束后栈中仍有元素, 则说明这些柱子右边没有比它更低的柱子了, 需要计算它们对应的矩形面积while stack:ch heights[stack.pop()]# 栈顶柱子的右边界r就是数组长度, 左边界l是上一个栈顶或-1(上一个栈顶不存在时)r len(heights)l -1 if not stack else stack[-1]# 宽*高res max(res, (r - l - 1) * ch)return res大家可以在下面这些地方找到我~ 我的 GitHub 我的 Leetcode 我的 CSDN 我的知乎专栏 我的头条号 我的牛客网博客 我的公众号: 算法精选, 欢迎大家扫码关注~
