代码随想录算法训练营第五十九天|503.下一个更大元素II、42.接雨水

代码随想录算法训练营第五十九天|503.下一个更大元素II、42.接雨水

题目

503.下一个更大元素II

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

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

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

题目

42.接雨水

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

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