代码随想录Day 51|leetcode|Python|503. 下一个更大元素 II，42. 接雨水

503. 下一个更大元素 II

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

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

解题思路：

对于循环数组，可以将遍历数组延长一倍，然后使用mod保证遍历数都在nums中，设置一个result长度为nums，设置stack存储数字，遍历每个数时遍历后面的数，分情况讨论，如果小于等于当前数，存入stack，大于的话更新result

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        result = [-1]*len(nums)
        stack = []
        stack.append(0)#initialize the stack put the first index in
        #iterate nums in nums length double
        for i in range(1, 2*len(nums)):
            i = i%(len(nums))
            if nums[i] <= nums[stack[-1]]:
                stack.append(i)
            else:
                while stack and nums[i]>nums[stack[-1]]:
                    result[stack[-1]] = nums[i]
                    stack.pop()
                stack.append(i)
        return result

 42. 接雨水

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

示例 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 个单位的雨水（蓝色部分表示雨水）。

解题思路：

运用单调栈步骤如下：

设置satck用来存储单调递增元素

将第一个元素index加入stack

遍历每一个元素并与栈顶进行对比，小于等于存入stack，大于的话作为边界之一，目前stack内部元素应该是递增的关系，因此弹出栈顶 bottom作为底部，新的栈顶作为边界之一

计算空间capacity height = min(nums[i], height(stack[-1])) - height [bottom]，宽度 width = i - stack[-1]- 1

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