Day59代码随想录单调栈part02:503.下一个更大元素II、42. 接雨水

Day59单调栈 part02

503.下一个更大元素II

leetcode题目链接:503. 下一个更大元素 II - 力扣(LeetCode)

题目:给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。

示例 1:

  • 输入: [1,2,1]
  • 输出: [2,-1,2]
  • 解释: 第一个 1 的下一个更大的数是 2;数字 2 找不到下一个更大的数;第二个 1 的下一个最大的数需要循环搜索,结果也是 2

思路:相比于之前的题目,现在是一个环

如何处理循环数组:将两个nums数组拼接在一起使用单调栈计算出每一个元素的下一个最大值,最后再把结果集即result数组resize到原数组大小就可以了。

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        res = [-1]*len(nums)
        st = [0]
        for i in range(1, len(nums)*2):
            if nums[st[-1]] >= nums[i%len(nums)]:
                st.append(i%len(nums))
            else:
                while st and nums[st[-1]] < nums[i%len(nums)]:
                    res[st[-1]] = nums[i%len(nums)]
                    st.pop()
                st.append(i%len(nums))
        return res

42. 接雨水

leetcode题目链接:. - 力扣(LeetCode)

题意:给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。

  • 输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
  • 输出:6
  • 解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。

思路:本题通常有暴力解法、双指针优化、单调栈三种解法

单调栈解法:接雨水这道题目,我们正需要寻找一个元素,右边最大元素(单调栈找到第一个右边最大)以及左边最大元素(单调栈st中的下一个元素,st[-2]如果现在是st[-1]的话),来计算雨水面积。

class Solution:
    def trap(self, height: List[int]) -> int:
        res = 0
        st = [0]
        for i in range(1, len(height)):
            if height[i] <= height[st[-1]]:
                st.append(i)
            else:
                while st and height[i] > height[st[-1]]:
                    mid_height = height[st[-1]]
                    st.pop()
                    if st:
                        h = min(height[i], height[st[-1]])
                        width = i-st[-1]-1
                        res += (h-mid_height)*width
                        # print(st)
                        # print(i, res, (h-mid_height)*width, h, mid_height, width)
                st.append(i)
        return res

  • 4
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值