leetcode 打卡 day60

84.柱状图中最大的矩形

双指针,超时

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        res = 0
        for i in range(len(heights)):
            left = i
            right = i
            # 向左侧遍历,寻找第一个矮一级的柱子
            for _ in range(left, -1, -1):
                if heights[left] < heights[i]:
                    break
                left -= 1
            for _ in range(right, len(heights)):
                if heights[right] < heights[i]:
                    break
                right += 1
            
            width = right - left - 1
            height = heights[i]
            res = max(res, width * height)
        return res

动态规划

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        # 动态规划
        length = len(heights)
        # 两个DP数列储存的均是下标index
        min_left_index = [0] * length
        min_right_index = [0] * length
        result = 0
        # 记录每个柱子的左侧第一个矮一级的柱子下标
        # 初始化防止while死循环
        min_left_index[0] = -1
        for i in range(1, length):
            # 以当前柱子为主心骨,向左迭代寻找次级柱子
            temp = i - 1
            while temp >= 0 and heights[temp] >= heights[i]:
                # 当左侧柱子持续较高时,尝试这个高柱子向左的比这个高柱子矮的柱子
                temp = min_left_index[temp]
            # 找到左侧矮的柱子
            min_left_index[i] = temp
        # 记录每个柱子右侧第一个矮一级的柱子下标
        # 初始化防止while死循环
        min_right_index[length-1] = length
        for i in range(length-2, -1, -1):
            temp = i + 1
            while temp < length and heights[temp] >= heights[i]:
                # 当右侧柱子持续较高时,尝试这个高柱子向左的比这个高柱子矮的柱子
                temp = min_right_index[temp]
            # 找到右侧矮一级柱子
            min_right_index[i] = temp
        
        for i in range(length):
            area = heights[i] * (min_right_index[i] - min_left_index[i] - 1)
            result = max(area, result)
        return result

单调栈
本题是要找每个柱子左右两边第一个小于该柱子的柱子,所以从栈头(元素从栈头弹出)到栈底的顺序应该是从大到小的顺序!与接雨水顺序相反。只有栈里从大到小的顺序,才能保证栈顶元素找到左右两边第一个小于栈顶元素的柱子。
栈顶和栈顶的下一个元素以及要入栈的三个元素组成了我们要求最大面积的高度和宽度。
剩下就是分析清楚如下三种情况:
情况一:当前遍历的元素heights[i]大于栈顶元素heights[st.top()]的情况
情况二:当前遍历的元素heights[i]等于栈顶元素heights[st.top()]的情况
情况三:当前遍历的元素heights[i]小于栈顶元素heights[st.top()]的情况

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        # 单调栈
        '''
        找每个柱子左右侧的第一个高度值小于该柱子的柱子
        单调栈:栈顶到栈底:从大到小(每插入一个新的小数值时,都要弹出先前的大数值)
        栈顶, 栈顶的下一个元素,即将入栈的元素:这三个元素组成了最大面积的高度和宽度
        情况一:当前遍历的元素heights[i]大于栈顶元素的情况
        情况二:当前遍历的元素heights[i]等于栈顶元素的情况
        情况三:当前遍历的元素heights[i]小于栈顶元素的情况
        '''
        # 输入数组首尾各补上一个0(本题原首尾的两个柱子可以作为核心柱进行最大面积尝试)
        heights.insert(0, 0)
        heights.append(0)
        stack = [0]
        result = 0
        for i in range(1, len(heights)):
            # 情况一
            if heights[i] > heights[stack[-1]]:
                stack.append(i)
            # 情况二
            elif heights[i] == heights[stack[-1]]:
                stack.pop()
                stack.append(i)
            else:
                # 抛出所有较高的柱子
                while stack and heights[i] < heights[stack[-1]]:
                    # 栈顶就是中间的柱子,主心骨
                    mid_index = stack[-1]
                    stack.pop()
                    if stack:
                        left_index = stack[-1]
                        right_index = i
                        width = right_index - left_index - 1
                        height = heights[mid_index]
                        result = max(result, width * height)
                stack.append(i)
        return result 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值