代码随想录算法训练营第五十九天|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